diff options
| author | wywywywy <wywywywy@gmail.com> | 2022-12-04 13:42:19 +0000 |
|---|---|---|
| committer | wywywywy <wywywywy@gmail.com> | 2022-12-04 13:42:19 +0000 |
| commit | a8ae263c695c1da8d05b2140e2225babcfb0a387 (patch) | |
| tree | e25baf7a555c3339f5ef9f69a4e892c612a02aba /extensions-builtin/LDSR/ldsr_model_arch.py | |
| parent | 44c46f0ed395967cd3830dd481a2db759fda5b3b (diff) | |
Reinstate DDPM V1 to LDSR
Diffstat (limited to 'extensions-builtin/LDSR/ldsr_model_arch.py')
| -rw-r--r-- | extensions-builtin/LDSR/ldsr_model_arch.py | 1 |
1 files changed, 1 insertions, 0 deletions
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() |