salahIguiliz commited on
Commit
3d87618
·
1 Parent(s): 19eac47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,