Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
|
49 |
# remove following line if xformers is not installed or when using Torch 2.0.
|
50 |
# pipe.enable_xformers_memory_efficient_attention()
|
51 |
# memory optimization.
|
52 |
-
pipe.enable_model_cpu_offload()
|
53 |
|
54 |
def resize_image(input_image, resolution):
|
55 |
H, W, C = input_image.shape
|
|
|
49 |
# remove following line if xformers is not installed or when using Torch 2.0.
|
50 |
# pipe.enable_xformers_memory_efficient_attention()
|
51 |
# memory optimization.
|
52 |
+
# pipe.enable_model_cpu_offload()
|
53 |
|
54 |
def resize_image(input_image, resolution):
|
55 |
H, W, C = input_image.shape
|