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