Divyanshu04 commited on
Commit
5366747
·
1 Parent(s): 522dbe7
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ login(token='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')
20
  pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-model", safety_checker=None, torch_dtype=torch.float32).to("cpu")
21
 
22
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
23
- pipe.enable_xformers_memory_efficient_attention()
24
  g_cuda = None
25
 
26
  FILE = Path(__file__).resolve()
 
20
  pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-model", safety_checker=None, torch_dtype=torch.float32).to("cpu")
21
 
22
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
23
+ # pipe.enable_xformers_memory_efficient_attention()
24
  g_cuda = None
25
 
26
  FILE = Path(__file__).resolve()