Update config.yml
Browse files- config.yml +3 -3
config.yml
CHANGED
@@ -100,9 +100,9 @@ train_ms:
|
|
100 |
# 注意, “:” 后需要加空格
|
101 |
webui:
|
102 |
# 推理设备
|
103 |
-
device: "
|
104 |
# 模型路径
|
105 |
-
model: "models/
|
106 |
# 配置文件路径
|
107 |
config_path: "config.json"
|
108 |
# 端口号
|
@@ -160,7 +160,7 @@ server:
|
|
160 |
# 模型config.json的路径
|
161 |
config: ""
|
162 |
# 模型使用设备,若填写则会覆盖默认配置
|
163 |
-
device: "
|
164 |
# 模型默认使用的语言
|
165 |
language: "JP"
|
166 |
# 模型人物默认参数
|
|
|
100 |
# 注意, “:” 后需要加空格
|
101 |
webui:
|
102 |
# 推理设备
|
103 |
+
device: "cuda"
|
104 |
# 模型路径
|
105 |
+
model: "models/G_8000.pth"
|
106 |
# 配置文件路径
|
107 |
config_path: "config.json"
|
108 |
# 端口号
|
|
|
160 |
# 模型config.json的路径
|
161 |
config: ""
|
162 |
# 模型使用设备,若填写则会覆盖默认配置
|
163 |
+
device: "cuda"
|
164 |
# 模型默认使用的语言
|
165 |
language: "JP"
|
166 |
# 模型人物默认参数
|