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

Jonas Müller · web-flow · commit e25e4c0df1d0 · 2023-07-21T18:00:53.000+02:00
Fix loading safetensors with load_model_from_config
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)
 
