Merge pull request Stability-AI#43 from akx/fix-safetensors-load

Fix loading safetensors with load_model_from_config
SevanBrodjian · Jul 21, 2023 · 14a2c23 · 14a2c23
2 parents a0e5865 + 7ebdbf5
commit 14a2c23
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)