Spaces:
Runtime error
Runtime error
maximuschan
commited on
Commit
•
6d3760e
1
Parent(s):
5e4cf3e
Update app.py
Browse files
app.py
CHANGED
@@ -70,6 +70,7 @@ config = XttsConfig()
|
|
70 |
config.load_json(os.path.join(model_path, "config.json"))
|
71 |
|
72 |
model = Xtts.init_from_config(config)
|
|
|
73 |
model.load_checkpoint(
|
74 |
config,
|
75 |
checkpoint_path=os.path.join(model_path, "model.pth"),
|
|
|
70 |
config.load_json(os.path.join(model_path, "config.json"))
|
71 |
|
72 |
model = Xtts.init_from_config(config)
|
73 |
+
model.cuda()
|
74 |
model.load_checkpoint(
|
75 |
config,
|
76 |
checkpoint_path=os.path.join(model_path, "model.pth"),
|