ylacombe HF staff commited on
Commit
b9ebfc8
1 Parent(s): ac1e9e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ model.load_checkpoint(
56
  checkpoint_path=os.path.join(model_path, "model.pth"),
57
  vocab_path=os.path.join(model_path, "vocab.json"),
58
  eval=True,
59
- use_deepspeed=False, # TODO: replace by True
60
  )
61
  model.cuda()
62
  print("Done loading TTS")
 
56
  checkpoint_path=os.path.join(model_path, "model.pth"),
57
  vocab_path=os.path.join(model_path, "vocab.json"),
58
  eval=True,
59
+ use_deepspeed=True,
60
  )
61
  model.cuda()
62
  print("Done loading TTS")