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