RamAnanth1 commited on
Commit
4cacdb2
1 Parent(s): a9d7f19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -243,7 +243,7 @@ class canny2image:
243
  # need to explicitly call pipe.to("cuda").
244
  self.pipe.enable_model_cpu_offload()
245
 
246
- self.pipe.enable_xformers_memory_efficient_attention()
247
 
248
  # Generator seed,
249
  self.generator = torch.manual_seed(0)
 
243
  # need to explicitly call pipe.to("cuda").
244
  self.pipe.enable_model_cpu_offload()
245
 
246
+ #self.pipe.enable_xformers_memory_efficient_attention()
247
 
248
  # Generator seed,
249
  self.generator = torch.manual_seed(0)