uix2345x commited on
Commit
0bb5a80
1 Parent(s): 861e32e

only free cpu device available

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