accessible-mistral

Paused

ylacombe HF staff commited on Oct 18, 2023

Commit

b9ebfc8

•

1 Parent(s): ac1e9e7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -56,7 +56,7 @@ model.load_checkpoint(
     checkpoint_path=os.path.join(model_path, "model.pth"),
     vocab_path=os.path.join(model_path, "vocab.json"),
     eval=True,
-    use_deepspeed=False, # TODO: replace by True
 )
 model.cuda()
 print("Done loading TTS")

     checkpoint_path=os.path.join(model_path, "model.pth"),
     vocab_path=os.path.join(model_path, "vocab.json"),
     eval=True,
+    use_deepspeed=True,
 )
 model.cuda()
 print("Done loading TTS")