Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,8 +24,8 @@ pipeline = DiffusionPipeline.from_pretrained(
24
  )
25
  pipeline.load_lora_weights(lora_weights_path)
26
 
27
- if device == "cuda":
28
- pipeline.enable_sequential_cpu_offload() # Efficient memory usage
29
 
30
  pipeline = pipeline.to(device)
31
 
 
24
  )
25
  pipeline.load_lora_weights(lora_weights_path)
26
 
27
+ # Comment out the line for sequential CPU offloading
28
+ # pipeline.enable_sequential_cpu_offload()
29
 
30
  pipeline = pipeline.to(device)
31