Fix loading safetensors with load_model_from_config

Previously, the `sd` from the safetensors if branch was not used at all, and `pl_sd` would have not been assigned.
SevanBrodjian · Jul 17, 2023 · 7ebdbf5 · 7ebdbf5
1 parent 90aef8c
commit 7ebdbf5
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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)