Spaces:
Sleeping
Sleeping
update app
Browse files
app.py
CHANGED
@@ -18,10 +18,11 @@ from pytorch_lightning import seed_everything
|
|
18 |
from torch import autocast
|
19 |
from contextlib import contextmanager, nullcontext
|
20 |
|
21 |
-
|
22 |
-
from
|
23 |
-
from
|
24 |
-
from
|
|
|
25 |
|
26 |
# from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
27 |
from transformers import AutoFeatureExtractor
|
|
|
18 |
from torch import autocast
|
19 |
from contextlib import contextmanager, nullcontext
|
20 |
|
21 |
+
sys.path.append("./stable-diffusion/")
|
22 |
+
from ldm.util import instantiate_from_config
|
23 |
+
from ldm.models.diffusion.psld import DDIMSampler
|
24 |
+
from ldm.models.diffusion.plms import PLMSSampler
|
25 |
+
from ldm.models.diffusion.dpm_solver import DPMSolverSampler
|
26 |
|
27 |
# from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
28 |
from transformers import AutoFeatureExtractor
|