From 1671967068d6a972c07e98cbd8bed56d5ed3a908 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Sat, 22 Jun 2024 17:24:21 +0900 Subject: [PATCH] Style --- train_ms.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train_ms.py b/train_ms.py index b1cc3a4..9f8ab5f 100644 --- a/train_ms.py +++ b/train_ms.py @@ -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.