Spaces:
Runtime error
Runtime error
TheFriendlyNPC
commited on
Commit
•
adf582b
1
Parent(s):
78d14e7
change to use cpu
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ models, cfg, task = load_model_ensemble_and_task_from_hf_hub(
|
|
40 |
)
|
41 |
|
42 |
model = models[0]
|
43 |
-
model = model.to(torch.device("cuda:0"))
|
44 |
|
45 |
TTSHubInterface.update_cfg_with_data_cfg(cfg, task.data_cfg)
|
46 |
generator = task.build_generator(models, cfg)
|
|
|
40 |
)
|
41 |
|
42 |
model = models[0]
|
43 |
+
model = model.to(torch.device("cuda:0")) if torch.cuda.is_available() else model
|
44 |
|
45 |
TTSHubInterface.update_cfg_with_data_cfg(cfg, task.data_cfg)
|
46 |
generator = task.build_generator(models, cfg)
|