Merge pull request #5415 from wywywywy/reinstate-ddpm-v1

Reinstate DDPM V1 to LDSR

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()

extensions-builtin/LDSR/scripts/ldsr_model.py

@@ -7,7 +7,7 @@ from basicsr.utils.download_util import load_file_from_url
 from modules.upscaler import Upscaler, UpscalerData
 from ldsr_model_arch import LDSR
 from modules import shared, script_callbacks
-import sd_hijack_autoencoder
+import sd_hijack_autoencoder, sd_hijack_ddpm_v1
 
 
 class UpscalerLDSR(Upscaler):