给配置文件功能加上is_half选项

chenguo5-tal · Apr 9, 2024 · d4c4b50 · d4c4b50
1 parent 27debe2
commit d4c4b50
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/config.json b/config.json
@@ -5,5 +5,6 @@
     "gpt_path":"./models/gpt.ckpt",
     "default_refer_path":"./models/referaudio.wav",
     "default_refer_text":"参考文本",
-    "default_refer_language":"zh"
+    "default_refer_language":"zh",
+    "is_half":false
 }
diff --git a/server.py b/server.py
@@ -35,6 +35,7 @@
     default_refer_path = config["default_refer_path"]
     default_refer_text = config["default_refer_text"]
     default_refer_language = config["default_refer_language"]
+    is_half = config["is_half"]
 # 自动判断环境是否支持CUDA和DirectML
 if(torch.cuda.is_available()):
     print("CUDA可用，将使用CUDA进行推理加速。")
@@ -48,7 +49,6 @@
         device = torch_directml.device(0)
         print("DirectML可用，将使用DirectML进行推理加速。")
         print("设备名称:",torch_directml.device_name(0))
-is_half = False
 # -----------------------
 
 # 如果要增加更多的参数选项，在这里设定