From 4a30b77b480d0ede92cc789e3ed1b13766254d25 Mon Sep 17 00:00:00 2001
From: Azalea <22280294+hykilpikonna@users.noreply.github.com>
Date: Sat, 14 Oct 2023 17:31:25 -0400
Subject: [PATCH] [O] Enable torch backward compatibility

---
 msclap/CLAPWrapper.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/msclap/CLAPWrapper.py b/msclap/CLAPWrapper.py
index c899be3..b75eb37 100644
--- a/msclap/CLAPWrapper.py
+++ b/msclap/CLAPWrapper.py
@@ -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)