xhlm123 commited on
Commit
e7eaa5e
·
verified ·
1 Parent(s): 8c04ed9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -24,23 +24,23 @@ if os.path.exists("./gweight.txt"):
24
  "gpt_path", gweight_data)
25
  else:
26
  gpt_path = os.environ.get(
27
- "gpt_path", "GPT_SoVITS/pretrained_models/hutao-e15.ckpt")
28
 
29
  if os.path.exists("./sweight.txt"):
30
  with open("./sweight.txt", 'r', encoding="utf-8") as file:
31
  sweight_data = file.read()
32
  sovits_path = os.environ.get("sovits_path", sweight_data)
33
  else:
34
- sovits_path = os.environ.get("sovits_path", "GPT_SoVITS/pretrained_models/hutao_e12_s1740.pth")
35
  # gpt_path = os.environ.get(
36
  # "gpt_path", "pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"
37
  # )
38
  # sovits_path = os.environ.get("sovits_path", "pretrained_models/s2G488k.pth")
39
  cnhubert_base_path = os.environ.get(
40
- "cnhubert_base_path", "GPT_SoVITS/pretrained_models/chinese-hubert-base"
41
  )
42
  bert_path = os.environ.get(
43
- "bert_path", "GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large"
44
  )
45
  infer_ttswebui = os.environ.get("infer_ttswebui", 9872)
46
  infer_ttswebui = int(infer_ttswebui)
@@ -584,8 +584,8 @@ def change_choices():
584
  return {"choices": sorted(SoVITS_names, key=custom_sort_key), "__type__": "update"}, {"choices": sorted(GPT_names, key=custom_sort_key), "__type__": "update"}
585
 
586
 
587
- pretrained_sovits_name = "GPT_SoVITS/pretrained_models/s2G488k.pth"
588
- pretrained_gpt_name = "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"
589
  SoVITS_weight_root = "SoVITS_weights"
590
  GPT_weight_root = "GPT_weights"
591
  os.makedirs(SoVITS_weight_root, exist_ok=True)
 
24
  "gpt_path", gweight_data)
25
  else:
26
  gpt_path = os.environ.get(
27
+ "gpt_path", "pretrained_models/hutao-e15.ckpt")
28
 
29
  if os.path.exists("./sweight.txt"):
30
  with open("./sweight.txt", 'r', encoding="utf-8") as file:
31
  sweight_data = file.read()
32
  sovits_path = os.environ.get("sovits_path", sweight_data)
33
  else:
34
+ sovits_path = os.environ.get("sovits_path", "pretrained_models/hutao_e12_s1740.pth")
35
  # gpt_path = os.environ.get(
36
  # "gpt_path", "pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"
37
  # )
38
  # sovits_path = os.environ.get("sovits_path", "pretrained_models/s2G488k.pth")
39
  cnhubert_base_path = os.environ.get(
40
+ "cnhubert_base_path", "pretrained_models/chinese-hubert-base"
41
  )
42
  bert_path = os.environ.get(
43
+ "bert_path", "pretrained_models/chinese-roberta-wwm-ext-large"
44
  )
45
  infer_ttswebui = os.environ.get("infer_ttswebui", 9872)
46
  infer_ttswebui = int(infer_ttswebui)
 
584
  return {"choices": sorted(SoVITS_names, key=custom_sort_key), "__type__": "update"}, {"choices": sorted(GPT_names, key=custom_sort_key), "__type__": "update"}
585
 
586
 
587
+ pretrained_sovits_name = "pretrained_models/hutao-e15.ckpt"
588
+ pretrained_gpt_name = "pretrained_models/hutao_e12_s1740.pth"
589
  SoVITS_weight_root = "SoVITS_weights"
590
  GPT_weight_root = "GPT_weights"
591
  os.makedirs(SoVITS_weight_root, exist_ok=True)