Update config.yml

config.yml

@@ -54,9 +54,9 @@ bert_gen:
   config_path: "config.json"
   # 并行数
   num_processes: 4
-  # 使用设备：可选项 "cpu" 显卡推理，"cpu" cpu推理
+  # 使用设备：可选项 "cuda" 显卡推理，"cuda" cuda推理
   # 该选项同时决定了get_bert_feature的默认设备
-  device: "cpu"
+  device: "cuda"
   # 使用多卡推理
   use_multi_device: false
 
@@ -67,8 +67,8 @@ emo_gen:
   config_path: "config.json"
   # 并行数
   num_processes: 4
-  # 使用设备：可选项 "cpu" 显卡推理，"cpu" cpu推理
-  device: "cpu"
+  # 使用设备：可选项 "cuda" 显卡推理，"cuda" cuda推理
+  device: "cuda"
   # 使用多卡推理
   use_multi_device: false
 
@@ -104,7 +104,7 @@ train_ms:
 # 注意， “:” 后需要加空格
 webui:
   # 推理设备
-  device: "cpu"
+  device: "cuda"
   # 模型路径
   model: "models/G.pth"
   # 配置文件路径
@@ -126,7 +126,7 @@ server:
   # 端口号
   port: 5000
   # 模型默认使用设备：但是当前并没有实现这个配置。
-  device: "cpu"
+  device: "cuda"
   # 需要加载的所有模型的配置，可以填多个模型，也可以不填模型，等网页成功后手动加载模型
   # 不加载模型的配置格式：删除默认给的两个模型配置，给models赋值 [ ]，也就是空列表。参考模型2的speakers 即 models: [ ]
   # 注意，所有模型都必须正确配置model与config的路径，空路径会导致加载错误。
@@ -137,7 +137,7 @@ server:
       # 模型config.json的路径
       config: ""
       # 模型使用设备，若填写则会覆盖默认配置
-      device: "cpu"
+      device: "cuda"
       # 模型默认使用的语言
       language: "ZH"
       # 模型人物默认参数
@@ -164,7 +164,7 @@ server:
       # 模型config.json的路径
       config: ""
       # 模型使用设备，若填写则会覆盖默认配置
-      device: "cpu"
+      device: "cuda"
       # 模型默认使用的语言
       language: "JP"
       # 模型人物默认参数