ritutweets46 Aditibaheti commited on
Commit
3f5fbd0
·
verified ·
1 Parent(s): 04f7a5c

Update app.py (#6)

Browse files

- Update app.py (fee5ba4502677679d7ce481ebbdf07cb3ed6c493)


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

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,8 +23,8 @@ pipeline = DiffusionPipeline.from_pretrained(
23
  use_auth_token=HUGGINGFACE_TOKEN
24
  )
25
  pipeline.load_lora_weights(lora_weights_path)
26
- pipeline.enable_sequential_cpu_offload() # Efficient memory usage
27
- pipeline.enable_xformers_memory_efficient_attention() # Enable xformers memory efficient attention
28
  pipeline = pipeline.to(device)
29
 
30
  MAX_SEED = np.iinfo(np.int32).max
 
23
  use_auth_token=HUGGINGFACE_TOKEN
24
  )
25
  pipeline.load_lora_weights(lora_weights_path)
26
+ #pipeline.enable_sequential_cpu_offload() # Efficient memory usage
27
+ #pipeline.enable_xformers_memory_efficient_attention() # Enable xformers memory efficient attention
28
  pipeline = pipeline.to(device)
29
 
30
  MAX_SEED = np.iinfo(np.int32).max