From dd1050cf82adc37d2c9f312a9f40d23f5c181751 Mon Sep 17 00:00:00 2001 From: lunar Date: Fri, 2 Dec 2022 12:17:13 +0000 Subject: [PATCH] small fixes --- example1_standard.py | 2 +- latent_blending.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) 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():