SpicyqSama007 commited on
Commit
426672b
1 Parent(s): 3f985d1

Update config.yml

Browse files
Files changed (1) hide show
  1. config.yml +1 -1
config.yml CHANGED
@@ -110,7 +110,7 @@ train_ms:
110
  # 注意, “:” 后需要加空格
111
  webui:
112
  # 推理设备
113
- device: "cuda"
114
  # 模型路径
115
  model: "models/G_3000_release.pth"
116
  # 配置文件路径
 
110
  # 注意, “:” 后需要加空格
111
  webui:
112
  # 推理设备
113
+ device: "cpu"
114
  # 模型路径
115
  model: "models/G_3000_release.pth"
116
  # 配置文件路径