Spaces:
Build error
Build error
Commit
·
3d87618
1
Parent(s):
19eac47
Update app.py
Browse files
app.py
CHANGED
@@ -85,7 +85,7 @@ def inference(prompt,canny_image,number,seed ):
|
|
85 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
86 |
# This command loads the individual model components on GPU on-demand. So, we don't
|
87 |
# need to explicitly call pipe.to("cuda").
|
88 |
-
pipe.enable_model_cpu_offload()
|
89 |
# xformers
|
90 |
pipe.enable_xformers_memory_efficient_attention()
|
91 |
# Generator seed,
|
|
|
85 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
86 |
# This command loads the individual model components on GPU on-demand. So, we don't
|
87 |
# need to explicitly call pipe.to("cuda").
|
88 |
+
#pipe.enable_model_cpu_offload()
|
89 |
# xformers
|
90 |
pipe.enable_xformers_memory_efficient_attention()
|
91 |
# Generator seed,
|