Spaces:
Runtime error
Runtime error
Enable torch.compile
Browse files
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)
|