small fixes

This commit is contained in:
lunar 2022-12-02 12:17:13 +00:00
parent 70ade8b73b
commit dd1050cf82
2 changed files with 4 additions and 4 deletions

View File

@ -31,7 +31,7 @@ from stable_diffusion_holder import StableDiffusionHolder
torch.set_grad_enabled(False) torch.set_grad_enabled(False)
#%% First let us spawn a stable diffusion holder #%% 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_ckpt = "../stable_diffusion_models/ckpt/768-v-ema.ckpt"
fp_config = '../stablediffusion/configs/stable-diffusion/v2-inference-v.yaml' fp_config = '../stablediffusion/configs/stable-diffusion/v2-inference-v.yaml'

View File

@ -14,8 +14,8 @@
import os, sys import os, sys
dp_git = "/home/lugo/git/" dp_git = "/home/lugo/git/"
sys.path.append(os.path.join(dp_git,'garden4'))
sys.path.append('util') sys.path.append('util')
# sys.path.append('../stablediffusion/ldm')
import torch import torch
torch.backends.cudnn.benchmark = False torch.backends.cudnn.benchmark = False
import numpy as np import numpy as np
@ -38,9 +38,9 @@ torch.set_grad_enabled(False)
from omegaconf import OmegaConf from omegaconf import OmegaConf
from torch import autocast from torch import autocast
from contextlib import nullcontext 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.models.diffusion.ddim import DDIMSampler
from ldm.util import instantiate_from_config
from stable_diffusion_holder import StableDiffusionHolder from stable_diffusion_holder import StableDiffusionHolder
#%% #%%
class LatentBlending(): class LatentBlending():