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

Update GPT_SoVITS/app_colab.py

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