[O] Enable torch backward compatibility

2023-10-14 17:31:25 -04:00
parent 8e6c8e632c
commit 4a30b77b48
1 changed files with 1 additions and 1 deletions
@@ -155,7 +155,7 @@ class CLAPWrapper():
                 args.num_layers, args.normalize_prefix, args.mapping_type, True, True)

        model_state_dict = torch.load(self.model_fp, map_location=torch.device('cpu'))['model']
-        clapcap.load_state_dict(model_state_dict)
+        clapcap.load_state_dict(model_state_dict, strict=False)

        clapcap.eval()  # set clap in eval mode
        tokenizer = AutoTokenizer.from_pretrained(args.text_model)