evBackend

Runtime error

Jofthomas HF staff commited on Aug 9

Commit

c2f6920

•

1 Parent(s): 9f9d8e2

Update coqui.py

Files changed (1) hide show

coqui.py CHANGED Viewed

@@ -88,7 +88,7 @@ model.load_checkpoint(
     checkpoint_dir=model_path,
     vocab_path=vocab_path,
     eval=True,
-    use_deepspeed=False,
 )
 model.cuda()

     checkpoint_dir=model_path,
     vocab_path=vocab_path,
     eval=True,
+    use_deepspeed=True,
 )
 model.cuda()