Spaces:
Runtime error
Runtime error
Commit
•
37d67d4
1
Parent(s):
9a7ba21
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ high_threshold = 200
|
|
13 |
# Models
|
14 |
controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
|
15 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
16 |
-
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
|
17 |
)
|
18 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
19 |
|
|
|
13 |
# Models
|
14 |
controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
|
15 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
16 |
+
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16
|
17 |
)
|
18 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
19 |
|