sd-webui/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/sd-webui/stable-diffusion-webui.git synced 2024-12-14 23:02:00 +03:00
Code Issues Projects Releases Wiki Activity

Added an empty personalization_config argument to instantiate_from_config when loading LDSR as it is a required argument. (#1722)

Browse Source
This commit is contained in:
Alejandro Gil 2022-12-22 14:14:21 -08:00 committed by GitHub
commit 9ffdf04172
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/sd_utils/__init__.py
View File

@ -944,7 +944,7 @@ class LDSR():
global_step = pl_sd["global_step"]
sd = pl_sd["state_dict"]
config = OmegaConf.load(self.yamlPath)
model = instantiate_from_config(config.model)
model = instantiate_from_config(config.model, personalization_config="")
m, u = model.load_state_dict(sd, strict=False)
model.cuda()
model.eval()