Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
08942fb
1
Parent(s):
0053e1b
Disable deepspeed usage in model checkpoint loading for improved compatibility
Browse files
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=
|
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
|