kevinwang676 commited on
Commit
15add2a
·
verified ·
1 Parent(s): a2c727f

Update GPT_SoVITS/app.py

Browse files
Files changed (1) hide show
  1. GPT_SoVITS/app.py +1 -2
GPT_SoVITS/app.py CHANGED
@@ -51,8 +51,7 @@ def get_pretrain_model_path(env_name, log_file, def_path):
51
  return pretrain_path
52
 
53
 
54
- # device = "cuda" if torch.cuda.is_available() else "cpu"
55
- device = 'cpu'
56
 
57
  gpt_path = get_pretrain_model_path('gpt_path', "./gweight.txt",
58
  "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt")
 
51
  return pretrain_path
52
 
53
 
54
+ device = "cuda" if torch.cuda.is_available() else "cpu"
 
55
 
56
  gpt_path = get_pretrain_model_path('gpt_path', "./gweight.txt",
57
  "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt")