This commit is contained in:
litagin02
2024-06-22 17:24:21 +09:00
parent e78d2a6f7e
commit 1671967068

View File

@@ -36,6 +36,7 @@ from style_bert_vits2.models.models import (
from style_bert_vits2.nlp.symbols import SYMBOLS
from style_bert_vits2.utils.stdout_wrapper import SAFE_STDOUT
torch.backends.cuda.matmul.allow_tf32 = True
torch.backends.cudnn.allow_tf32 = (
True # If encontered training problem,please try to disable TF32.