Update coqui.py
Browse files
coqui.py
CHANGED
|
@@ -88,7 +88,7 @@ model.load_checkpoint(
|
|
| 88 |
checkpoint_dir=model_path,
|
| 89 |
vocab_path=vocab_path,
|
| 90 |
eval=True,
|
| 91 |
-
use_deepspeed=
|
| 92 |
)
|
| 93 |
model.cuda()
|
| 94 |
|
|
|
|
| 88 |
checkpoint_dir=model_path,
|
| 89 |
vocab_path=vocab_path,
|
| 90 |
eval=True,
|
| 91 |
+
use_deepspeed=True,
|
| 92 |
)
|
| 93 |
model.cuda()
|
| 94 |
|