maximuschan commited on
Commit
6d3760e
1 Parent(s): 5e4cf3e

Update app.py

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