改成cpu运行

app.py

@@ -516,7 +516,7 @@ def init_webui(unet, vae, normal_t):
                     # 选择sampler
                     sampler_name = gr.Dropdown(["DDIM"], label="sampler", value="DDIM")  # , "euler a", "dpmpp 2m"
                     # 创建滑动条组件
-                    step = gr.Slider(minimum=1, maximum=1000, value=400, label="步长", step=1)
+                    step = gr.Slider(minimum=1, maximum=1000, value=40, label="步长", step=1)
                     batch_size = gr.Slider(minimum=1, maximum=4, label="batch size", step=1)
                     img_size = gr.Slider(minimum=256, maximum=512, value=256, label="img size", step=64)
                     # 创建开始按钮组件
@@ -531,7 +531,7 @@ def init_webui(unet, vae, normal_t):
 
 
 if __name__ == '__main__':
-    device = "cpu"
+    device = torch.device('cpu')
     unet, vae, normal_t = get_models(device)