MohamedRashad commited on
Commit
08942fb
·
1 Parent(s): 0053e1b

Disable deepspeed usage in model checkpoint loading for improved compatibility

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import spaces
2
- import deepspeed
3
  import torch
4
  from TTS.tts.configs.xtts_config import XttsConfig
5
  from TTS.tts.models.xtts import Xtts
@@ -39,7 +38,7 @@ print("Loading model...")
39
  device = "cuda" if torch.cuda.is_available() else "cpu"
40
  print(device)
41
  model = Xtts.init_from_config(config)
42
- model.load_checkpoint(config, checkpoint_path=model_path, use_deepspeed=True, vocab_path=vocab_path, eval=True)
43
  model.to(device)
44
 
45
  @spaces.GPU
 
1
  import spaces
 
2
  import torch
3
  from TTS.tts.configs.xtts_config import XttsConfig
4
  from TTS.tts.models.xtts import Xtts
 
38
  device = "cuda" if torch.cuda.is_available() else "cpu"
39
  print(device)
40
  model = Xtts.init_from_config(config)
41
+ model.load_checkpoint(config, checkpoint_path=model_path, use_deepspeed=False, vocab_path=vocab_path, eval=True)
42
  model.to(device)
43
 
44
  @spaces.GPU