qxdn commited on
Commit
a1c2632
1 Parent(s): 00283c8

Update config.yml

Browse files
Files changed (1) hide show
  1. config.yml +1 -1
config.yml CHANGED
@@ -98,7 +98,7 @@ train_ms:
98
  # 注意, “:” 后需要加空格
99
  webui:
100
  # 推理设备
101
- device: "cuda"
102
  # 模型路径
103
  model: "models/G_120000.pth"
104
  # 配置文件路径
 
98
  # 注意, “:” 后需要加空格
99
  webui:
100
  # 推理设备
101
+ device: "cpu"
102
  # 模型路径
103
  model: "models/G_120000.pth"
104
  # 配置文件路径