small fixes
This commit is contained in:
parent
70ade8b73b
commit
dd1050cf82
|
@ -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'
|
||||||
|
|
||||||
|
|
|
@ -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():
|
||||||
|
|
Loading…
Reference in New Issue