juxuan27 commited on
Commit
fae6419
1 Parent(s): 8d3b536

Update app.py

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