From 41e804a38028ce9066f6a4e47f3f21f8a01c1d32 Mon Sep 17 00:00:00 2001 From: ZeroCool940711 Date: Thu, 22 Dec 2022 15:12:59 -0700 Subject: [PATCH] Added an empty `personalization_config` argument to `instantiate_from_config` when loading LDSR as it is a required argument. --- scripts/sd_utils/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/sd_utils/__init__.py b/scripts/sd_utils/__init__.py index 0d8b3e0..66977b9 100644 --- a/scripts/sd_utils/__init__.py +++ b/scripts/sd_utils/__init__.py @@ -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()