diff --git a/example1_standard.py b/example1_standard.py index 4af13d4..f8909e6 100644 --- a/example1_standard.py +++ b/example1_standard.py @@ -31,7 +31,7 @@ from stable_diffusion_holder import StableDiffusionHolder torch.set_grad_enabled(False) #%% First let us spawn a stable diffusion holder -device = "cuda:0" +device = "cuda:0" fp_ckpt = "../stable_diffusion_models/ckpt/768-v-ema.ckpt" fp_config = '../stablediffusion/configs/stable-diffusion/v2-inference-v.yaml' diff --git a/latent_blending.py b/latent_blending.py index 1a28358..a54b4aa 100644 --- a/latent_blending.py +++ b/latent_blending.py @@ -14,8 +14,8 @@ import os, sys dp_git = "/home/lugo/git/" -sys.path.append(os.path.join(dp_git,'garden4')) sys.path.append('util') +# sys.path.append('../stablediffusion/ldm') import torch torch.backends.cudnn.benchmark = False import numpy as np @@ -38,9 +38,9 @@ torch.set_grad_enabled(False) from omegaconf import OmegaConf from torch import autocast from contextlib import nullcontext -sys.path.append('../stablediffusion/ldm') -from ldm.util import instantiate_from_config + from ldm.models.diffusion.ddim import DDIMSampler +from ldm.util import instantiate_from_config from stable_diffusion_holder import StableDiffusionHolder #%% class LatentBlending():