5090不支持?
package = torch.load(model_path, map_location=lambda storage, loc: storage, weights_only=False)
然后:
File "/root/FireRedASR/fireredasr/models/fireredasr.py", line 111, in load_fireredasr_aed_model package = torch.load(model_path, map_location=lambda storage, loc: storage, weights_only=False) File "/root/miniconda3/envs/fireredasr/lib/python3.10/site-packages/torch/serialization.py", line 1554, in load return _legacy_load( File "/root/miniconda3/envs/fireredasr/lib/python3.10/site-packages/torch/serialization.py", line 1792, in _legacy_load return legacy_load(f) File "/root/miniconda3/envs/fireredasr/lib/python3.10/site-packages/torch/serialization.py", line 1663, in legacy_load tar.extract("storages", path=tmpdir) File "/root/miniconda3/envs/fireredasr/lib/python3.10/tarfile.py", line 2372, in extract tarinfo, unfiltered = self._get_extract_tarinfo( File "/root/miniconda3/envs/fireredasr/lib/python3.10/tarfile.py", line 2386, in _get_extract_tarinfo unfiltered = self.getmember(member) File "/root/miniconda3/envs/fireredasr/lib/python3.10/tarfile.py", line 2034, in getmember raise KeyError("filename %r not found" % name) KeyError: "filename 'storages' not found"