azalea/CLAP
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[O] Enable torch backward compatibility

Browse Source
This commit is contained in:
Azalea Gui
2023-10-14 17:31:25 -04:00
parent 8e6c8e632c
commit 4a30b77b48
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
msclap/CLAPWrapper.py
+1 -1
View File
@@ -155,7 +155,7 @@ class CLAPWrapper():
args.num_layers, args.normalize_prefix, args.mapping_type, True, True) 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'] 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 clapcap.eval() # set clap in eval mode
tokenizer = AutoTokenizer.from_pretrained(args.text_model) tokenizer = AutoTokenizer.from_pretrained(args.text_model)