Fabrice-TIERCELIN
commited on
Commit
•
e3aef47
1
Parent(s):
7969dfc
botp/stable-diffusion-v1-5
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ else:
|
|
26 |
|
27 |
controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11e_sd15_ip2p", torch_dtype = floatType)
|
28 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
29 |
-
"
|
30 |
)
|
31 |
pipe = pipe.to(device)
|
32 |
|
|
|
26 |
|
27 |
controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11e_sd15_ip2p", torch_dtype = floatType)
|
28 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
29 |
+
"botp/stable-diffusion-v1-5", safety_checker = None, controlnet = controlnet, torch_dtype = floatType
|
30 |
)
|
31 |
pipe = pipe.to(device)
|
32 |
|