L4Walk commited on
Commit
f126c88
·
1 Parent(s): aa43190

修改模型精度为32

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.float16,
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