修改模型精度为32
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
|
35 |
"runwayml/stable-diffusion-v1-5",
|
36 |
controlnet=controlnet,
|
37 |
safety_checker=None,
|
38 |
-
torch_dtype=torch.
|
39 |
).to("cpu")
|
40 |
# pipe.enable_xformers_memory_efficient_attention() # 只限GPU调用
|
41 |
|
|
|
35 |
"runwayml/stable-diffusion-v1-5",
|
36 |
controlnet=controlnet,
|
37 |
safety_checker=None,
|
38 |
+
torch_dtype=torch.float32,
|
39 |
).to("cpu")
|
40 |
# pipe.enable_xformers_memory_efficient_attention() # 只限GPU调用
|
41 |
|