Hugging Face's logo

Spaces:
Ryukijano
/
Fastest-image-generation
Running on Zero

App Files Community
Ryukijano commited on
Commit
638a50c
·
verified ·
1 Parent(s): 4ac4f4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ pipe.unload_lora_weights()
42
 
43
  # Memory optimizations
44
  pipe.transformer.to(memory_format=torch.channels_last)
45
- pipe.enable_xformers_memory_efficient_attention()
46
 
47
  # CUDA Graph setup
48
  static_inputs = None
 
42
 
43
  # Memory optimizations
44
  pipe.transformer.to(memory_format=torch.channels_last)
45
+ pipe.enable_attention_slicing()
46
 
47
  # CUDA Graph setup
48
  static_inputs = None