Reinstate DDPM V1 to LDSR

2025-08-08 13:19:54 +00:00 · 2022-12-04 13:42:19 +00:00
parent 44c46f0ed3
commit a8ae263c69
3 changed files with 1451 additions and 1 deletions
--- a/extensions-builtin/LDSR/ldsr_model_arch.py
+++ b/extensions-builtin/LDSR/ldsr_model_arch.py
@@ -22,6 +22,7 @@ class LDSR:
        pl_sd = torch.load(self.modelPath, map_location="cpu")
        sd = pl_sd["state_dict"]
        config = OmegaConf.load(self.yamlPath)
+        config.model.target = "ldm.models.diffusion.ddpm.LatentDiffusionV1"
        model = instantiate_from_config(config.model)
        model.load_state_dict(sd, strict=False)
        model.cuda()