Ming Li commited on
Commit
ea0de4a
1 Parent(s): 7238e55

update requirements.txt

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -59,7 +59,7 @@ class Model:
59
  )
60
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
61
  if self.device.type == "cuda":
62
- pipe.enable_xformers_memory_efficient_attention()
63
  pipe.to(self.device)
64
  torch.cuda.empty_cache()
65
  gc.collect()
 
59
  )
60
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
61
  if self.device.type == "cuda":
62
+ pipe.disable_xformers_memory_efficient_attention()
63
  pipe.to(self.device)
64
  torch.cuda.empty_cache()
65
  gc.collect()