Eugeoter commited on
Commit
b6eeb36
·
1 Parent(s): 067bc73
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ CACHE_DIR = None
16
  @spaces.GPU
17
  def optimize_pipeline(pipeline):
18
  pipeline.to(dtype=torch.float16)
19
- pipeline.enable_xformers_memory_efficient_attention()
20
  return pipeline
21
 
22
 
 
16
  @spaces.GPU
17
  def optimize_pipeline(pipeline):
18
  pipeline.to(dtype=torch.float16)
19
+ # pipeline.enable_xformers_memory_efficient_attention()
20
  return pipeline
21
 
22