Spaces:

Luminia
/

audiosep

Running

Nekochu commited on Jan 17

Commit

0c1d35f

verified ·

1 Parent(s): b5035e3

Upload models/CLAP/open_clip/factory.py with huggingface_hub

Files changed (1) hide show

models/CLAP/open_clip/factory.py CHANGED Viewed

@@ -150,7 +150,7 @@ def create_model(
                     f"Loading pretrained {amodel_name}-{tmodel_name} weights ({pretrained})."
                 )
                 ckpt = load_state_dict(checkpoint_path, skip_params=True)
-                model.load_state_dict(ckpt)
                 param_names = [n for n, p in model.named_parameters()]
                 # for n in param_names:
                 #     print(n, "\t", "Loaded" if n in ckpt else "Unloaded")

                     f"Loading pretrained {amodel_name}-{tmodel_name} weights ({pretrained})."
                 )
                 ckpt = load_state_dict(checkpoint_path, skip_params=True)
+                model.load_state_dict(ckpt, strict=False)
                 param_names = [n for n, p in model.named_parameters()]
                 # for n in param_names:
                 #     print(n, "\t", "Loaded" if n in ckpt else "Unloaded")