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

Reinstate DDPM V1 to LDSR
This commit is contained in:
AUTOMATIC1111 2022-12-10 09:20:38 +03:00 committed by GitHub
commit 3896242e9e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 1451 additions and 1 deletions

View File

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

View File

@ -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):

File diff suppressed because it is too large Load Diff