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

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

Browse Source 
* Update requirements.txt

* Update environment.yaml

* Update README.md

* Update environment.yaml

* Update README.md

* Update README.md

* Delete requirements_colossalai.txt

* Update requirements.txt

* Update README.md
This commit is contained in:
NatalieC323
2023-04-06 17:50:52 +08:00
committed by GitHub
parent 4e9989344d
commit c701b77b11
14 changed files with 124 additions and 98 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 main import instantiate_from_config
from ldm.models.diffusion.ddpm import LatentgDiffusion
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 = instantiate_from_config(config.model)
model = LatentDiffusion(**config.model.get("params", dict()))
model.load_state_dict(torch.load("models/ldm/inpainting_big/last.ckpt")["state_dict"],
strict=False)