Diffusers
3v324v23 commited on
Commit
8ad15f2
·
1 Parent(s): db1c8aa

added requirements memory speed xformers

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -119,8 +119,9 @@ class EndpointHandler:
119
  )
120
 
121
  # boosts performance by another 20%
122
- # self.pipe.enable_xformers_memory_efficient_attention()
123
- # self.pipe.enable_attention_slicing()
 
124
 
125
  def load_lora(self, pipeline, lora_path, lora_weight=0.5):
126
  state_dict = load_file(lora_path)
 
119
  )
120
 
121
  # boosts performance by another 20%
122
+ self.pipe.enable_xformers_memory_efficient_attention()
123
+ self.pipe.enable_attention_slicing()
124
+ # may need a requirement in the root with xformer
125
 
126
  def load_lora(self, pipeline, lora_path, lora_weight=0.5):
127
  state_dict = load_file(lora_path)