diff --git a/sgm/util.py b/sgm/util.py index 06f48a88..4820978f 100644 --- a/sgm/util.py +++ b/sgm/util.py @@ -212,7 +212,6 @@ def load_model_from_config(config, ckpt, verbose=True, freeze=True): raise NotImplementedError model = instantiate_from_config(config.model) - sd = pl_sd["state_dict"] m, u = model.load_state_dict(sd, strict=False)