fix

fireredasr/fireredasr/models/fireredasr.py

@@ -107,7 +107,7 @@ class FireRedAsr:
 
 
 def load_fireredasr_aed_model(model_path):
-    package = torch.load(model_path, map_location=lambda storage, loc: storage, weights_only=True)
+    package = torch.load(model_path, map_location=lambda storage, loc: storage, weights_only=False)
     print("model args:", package["args"])
     model = FireRedAsrAed.from_args(package["args"])
     model.load_state_dict(package["model_state_dict"], strict=True)
@@ -115,7 +115,7 @@ def load_fireredasr_aed_model(model_path):
 
 
 def load_firered_llm_model_and_tokenizer(model_path, encoder_path, llm_dir):
-    package = torch.load(model_path, map_location=lambda storage, loc: storage, weights_only=True)
+    package = torch.load(model_path, map_location=lambda storage, loc: storage, weights_only=False)
     package["args"].encoder_path = encoder_path
     package["args"].llm_dir = llm_dir
     print("model args:", package["args"])