Yaron Koresh commited on
Commit
66fc53c
·
verified ·
1 Parent(s): 582c5a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -17,6 +17,9 @@ from diffusers import DiffusionPipeline, StableDiffusionXLImg2ImgPipeline
17
  from diffusers.utils import load_image
18
  #import jax
19
  #import jax.numpy as jnp
 
 
 
20
 
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22
 
 
17
  from diffusers.utils import load_image
18
  #import jax
19
  #import jax.numpy as jnp
20
+ import torch._dynamo
21
+
22
+ torch._dynamo.config.suppress_errors = True
23
 
24
  device = "cuda" if torch.cuda.is_available() else "cpu"
25