Spaces:

ChenoAi
/

OpenDalle-demo

Runtime error

ChenoAi commited on Jun 1

Commit

5957a7d

•

1 Parent(s): 4addd7e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import numpy as np
 from PIL import Image
 import spaces
 import torch
-from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"

 from PIL import Image
 import spaces
 import torch
+from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler, AutoencoderKL
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"