Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -16,6 +16,7 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
|
16 |
controlnet=controlnet,
|
17 |
torch_dtype=torch.float16,
|
18 |
)
|
|
|
19 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
20 |
pipe.enable_model_cpu_offload()
|
21 |
generators = []
|
|
|
16 |
controlnet=controlnet,
|
17 |
torch_dtype=torch.float16,
|
18 |
)
|
19 |
+
pipe.to("cpu")
|
20 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
21 |
pipe.enable_model_cpu_offload()
|
22 |
generators = []
|