mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
extensions-builtin/LDSR/ldsr_model_arch.py
View File

@ -22,6 +22,7 @@ class LDSR:
pl_sd = torch.load(self.modelPath, map_location="cpu") pl_sd = torch.load(self.modelPath, map_location="cpu")
sd = pl_sd["state_dict"] sd = pl_sd["state_dict"]
config = OmegaConf.load(self.yamlPath) config = OmegaConf.load(self.yamlPath)
config.model.target = "ldm.models.diffusion.ddpm.LatentDiffusionV1"
model = instantiate_from_config(config.model) model = instantiate_from_config(config.model)
model.load_state_dict(sd, strict=False) model.load_state_dict(sd, strict=False)
model.cuda() model.cuda()

2
extensions-builtin/LDSR/scripts/ldsr_model.py
View File

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

1449
extensions-builtin/LDSR/sd_hijack_ddpm_v1.py Normal file
View File

File diff suppressed because it is too large Load Diff