github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-11 13:59:08 +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/inpaint.py
View File

@@ -4,7 +4,7 @@ from PIL import Image
from tqdm import tqdm
import numpy as np
import torch
from ldm.models.diffusion.ddpm import LatentgDiffusion
from main import instantiate_from_config
from ldm.models.diffusion.ddim import DDIMSampler
@@ -57,7 +57,7 @@ if __name__ == "__main__":
print(f"Found {len(masks)} inputs.")
config = OmegaConf.load("models/ldm/inpainting_big/config.yaml")
model = LatentDiffusion(**config.model.get("params", dict()))
model = instantiate_from_config(config.model)
model.load_state_dict(torch.load("models/ldm/inpainting_big/last.ckpt")["state_dict"],
strict=False)