Spaces:
Sleeping
Sleeping
upd
Browse files
app.py
CHANGED
|
@@ -34,6 +34,7 @@ def process_image(input_image: Image.Image, to_grayscale: bool, prompt: str) ->
|
|
| 34 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
| 35 |
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
|
| 36 |
)
|
|
|
|
| 37 |
|
| 38 |
output = pipe(
|
| 39 |
[prompt],
|
|
|
|
| 34 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
| 35 |
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
|
| 36 |
)
|
| 37 |
+
pipe.enable_model_cpu_offload()
|
| 38 |
|
| 39 |
output = pipe(
|
| 40 |
[prompt],
|