github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-10 13:30:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[dreambooth] fixing the incompatibity in requirements.txt (#3190) (#3378)" (#3481)

Browse Source
This commit is contained in:
NatalieC323
2023-04-06 20:22:52 +08:00
committed by GitHub
parent 891b8e7fac
commit fb8fae6f29
14 changed files with 98 additions and 124 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/images/diffusion/scripts/img2img.py
View File

@@ -20,8 +20,8 @@ from imwatermark import WatermarkEncoder
from scripts.txt2img import put_watermark
from ldm.util import instantiate_from_config
from ldm.models.diffusion.ddim import DDIMSampler
from ldm.models.diffusion.ddpm import LatentDiffusion
from utils import replace_module, getModelSize
@@ -36,7 +36,7 @@ def load_model_from_config(config, ckpt, verbose=False):
if "global_step" in pl_sd:
print(f"Global Step: {pl_sd['global_step']}")
sd = pl_sd["state_dict"]
model = LatentDiffusion(**config.model.get("params", dict()))
model = instantiate_from_config(config.model)
m, u = model.load_state_dict(sd, strict=False)
if len(m) > 0 and verbose:
print("missing keys:")