Spaces:

lord-reso
/

host

Sleeping

lord-reso commited on Feb 1, 2024

Commit

e24bfba

verified ·

1 Parent(s): d08438b

Update train.py

Files changed (1) hide show

train.py CHANGED Viewed

@@ -5,7 +5,7 @@ from model import Tacotron2
 from hparams import create_hparams
 def load_model(hparams):
-    model = Tacotron2(hparams).float()
     if hparams.fp16_run:
         model.decoder.attention_layer.score_mask_value = finfo('float16').min

 from hparams import create_hparams
 def load_model(hparams):
+    model = Tacotron2(hparams)
     if hparams.fp16_run:
         model.decoder.attention_layer.score_mask_value = finfo('float16').min