LituRout commited on
Commit
379f6d3
1 Parent(s): cff1d1d

update app

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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
- from stable-diffusion.ldm.util import instantiate_from_config
22
- from stable-diffusion.ldm.models.diffusion.psld import DDIMSampler
23
- from stable-diffusion.ldm.models.diffusion.plms import PLMSSampler
24
- from stable-diffusion.ldm.models.diffusion.dpm_solver import DPMSolverSampler
 
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