lnyan commited on
Commit
55b1429
1 Parent(s): 2f0048a

Enable torch.compile

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -308,6 +308,7 @@ class StableDiffusionInpaint:
308
  inpaint.tokenizer,
309
  )
310
  inpaint.to(device)
 
311
  # inpaint.enable_xformers_memory_efficient_attention()
312
  # if device == "mps":
313
  # _ = text2img("", num_inference_steps=1)
 
308
  inpaint.tokenizer,
309
  )
310
  inpaint.to(device)
311
+ inpaint=torch.compile(inpaint)
312
  # inpaint.enable_xformers_memory_efficient_attention()
313
  # if device == "mps":
314
  # _ = text2img("", num_inference_steps=1)