Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -24,6 +24,7 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
|
24 |
|
25 |
# speed up diffusion process with faster scheduler and memory optimization
|
26 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
|
|
27 |
#pipe.enable_model_cpu_offload()
|
28 |
#pipe = pipe.to("cuda")
|
29 |
|
|
|
24 |
|
25 |
# speed up diffusion process with faster scheduler and memory optimization
|
26 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
27 |
+
pipe.safety_checker = None
|
28 |
#pipe.enable_model_cpu_offload()
|
29 |
#pipe = pipe.to("cuda")
|
30 |
|