lnyan commited on
Commit
2296d8a
1 Parent(s): 43011fa
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -308,8 +308,11 @@ class StableDiffusionInpaint:
308
  inpaint.tokenizer,
309
  )
310
  inpaint.to(device)
311
- inpaint.vae=torch.compile(inpaint.vae, dynamic=True)
312
- inpaint.unet=torch.compile(inpaint.unet, dynamic=True)
 
 
 
313
  # inpaint.enable_xformers_memory_efficient_attention()
314
  # if device == "mps":
315
  # _ = text2img("", num_inference_steps=1)
 
308
  inpaint.tokenizer,
309
  )
310
  inpaint.to(device)
311
+ # try:
312
+ # inpaint.vae=torch.compile(inpaint.vae, dynamic=True)
313
+ # inpaint.unet=torch.compile(inpaint.unet, dynamic=True)
314
+ # except Exception as e:
315
+ # print(e)
316
  # inpaint.enable_xformers_memory_efficient_attention()
317
  # if device == "mps":
318
  # _ = text2img("", num_inference_steps=1)