From a8ae263c695c1da8d05b2140e2225babcfb0a387 Mon Sep 17 00:00:00 2001 From: wywywywy Date: Sun, 4 Dec 2022 13:42:19 +0000 Subject: Reinstate DDPM V1 to LDSR --- extensions-builtin/LDSR/ldsr_model_arch.py | 1 + 1 file changed, 1 insertion(+) (limited to 'extensions-builtin/LDSR/ldsr_model_arch.py') diff --git a/extensions-builtin/LDSR/ldsr_model_arch.py b/extensions-builtin/LDSR/ldsr_model_arch.py index 90e0a2f0..a87d1ef9 100644 --- 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() -- cgit v1.2.1