XzJosh commited on
Commit
3090446
1 Parent(s): 8fbdb1c

Update config.yml

Browse files
Files changed (1) hide show
  1. config.yml +1 -1
config.yml CHANGED
@@ -84,7 +84,7 @@ train_ms:
84
  # 注意, “:” 后需要加空格
85
  webui:
86
  # 推理设备
87
- device: "cuda"
88
  # 模型路径
89
  model: "models/G_7600.pth"
90
  # 配置文件路径
 
84
  # 注意, “:” 后需要加空格
85
  webui:
86
  # 推理设备
87
+ device: "cpu"
88
  # 模型路径
89
  model: "models/G_7600.pth"
90
  # 配置文件路径