Spaces:
Runtime error
Runtime error
Update GPT_SoVITS/app_colab.py
Browse files- 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 |
-
|
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")
|