diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-12-10 06:20:38 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-10 06:20:38 +0000 |
| commit | 3896242e9ecfe46c99bacdd1d590220f2510ebca (patch) | |
| tree | c72d64340017edc6f754047721d52fdcb27f61ca /extensions-builtin/LDSR/ldsr_model_arch.py | |
| parent | 505ec7e4d960e7bea579182509050fafb10bd00c (diff) | |
| parent | a8ae263c695c1da8d05b2140e2225babcfb0a387 (diff) | |
| download | stable-diffusion-webui-gfx803-3896242e9ecfe46c99bacdd1d590220f2510ebca.tar.gz stable-diffusion-webui-gfx803-3896242e9ecfe46c99bacdd1d590220f2510ebca.tar.bz2 stable-diffusion-webui-gfx803-3896242e9ecfe46c99bacdd1d590220f2510ebca.zip | |
Merge pull request #5415 from wywywywy/reinstate-ddpm-v1
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() |