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],
|