ritutweets46 Aditibaheti commited on
Commit
48e7986
1 Parent(s): a07a7b2

Update app.py (#14)

Browse files

- Update app.py (1681c1c6e3f4c0f136074c7d4c1b2c627054ce40)


Co-authored-by: Aditi Baheti <Aditibaheti@users.noreply.huggingface.co>

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