TheFriendlyNPC commited on
Commit
adf582b
1 Parent(s): 78d14e7

change to use cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)