lnyan commited on
Commit
43011fa
1 Parent(s): cf55df8
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -308,7 +308,8 @@ class StableDiffusionInpaint:
308
  inpaint.tokenizer,
309
  )
310
  inpaint.to(device)
311
- inpaint=torch.compile(inpaint, dynamic=True)
 
312
  # inpaint.enable_xformers_memory_efficient_attention()
313
  # if device == "mps":
314
  # _ = text2img("", num_inference_steps=1)
 
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)