torchaudio 2.4.1__cp312-cp312-manylinux2014_aarch64.whl → 2.5.1__cp312-cp312-manylinux2014_aarch64.whl

torchaudio/pipelines/_squim_pipeline.py

@@ -1,6 +1,7 @@
 from dataclasses import dataclass
 
-from torchaudio._internal import load_state_dict_from_url
+import torch
+import torchaudio
 
 from torchaudio.models import squim_objective_base, squim_subjective_base, SquimObjective, SquimSubjective
 
@@ -42,26 +43,16 @@ class SquimObjectiveBundle:
     _path: str
     _sample_rate: float
 
-    def _get_state_dict(self, dl_kwargs):
-        url = f"https://download.pytorch.org/torchaudio/models/{self._path}"
-        dl_kwargs = {} if dl_kwargs is None else dl_kwargs
-        state_dict = load_state_dict_from_url(url, **dl_kwargs)
-        return state_dict
-
-    def get_model(self, *, dl_kwargs=None) -> SquimObjective:
+    def get_model(self) -> SquimObjective:
         """Construct the SquimObjective model, and load the pretrained weight.
 
-        The weight file is downloaded from the internet and cached with
-        :func:`torch.hub.load_state_dict_from_url`
-
-        Args:
-            dl_kwargs (dictionary of keyword arguments): Passed to :func:`torch.hub.load_state_dict_from_url`.
-
         Returns:
             Variation of :py:class:`~torchaudio.models.SquimObjective`.
         """
         model = squim_objective_base()
-        model.load_state_dict(self._get_state_dict(dl_kwargs))
+        path = torchaudio.utils.download_asset(f"models/{self._path}")
+        state_dict = torch.load(path, weights_only=True)
+        model.load_state_dict(state_dict)
         model.eval()
         return model
 
@@ -128,26 +119,15 @@ class SquimSubjectiveBundle:
     _path: str
     _sample_rate: float
 
-    def _get_state_dict(self, dl_kwargs):
-        url = f"https://download.pytorch.org/torchaudio/models/{self._path}"
-        dl_kwargs = {} if dl_kwargs is None else dl_kwargs
-        state_dict = load_state_dict_from_url(url, **dl_kwargs)
-        return state_dict
-
-    def get_model(self, *, dl_kwargs=None) -> SquimSubjective:
+    def get_model(self) -> SquimSubjective:
         """Construct the SquimSubjective model, and load the pretrained weight.
-
-        The weight file is downloaded from the internet and cached with
-        :func:`torch.hub.load_state_dict_from_url`
-
-        Args:
-            dl_kwargs (dictionary of keyword arguments): Passed to :func:`torch.hub.load_state_dict_from_url`.
-
         Returns:
             Variation of :py:class:`~torchaudio.models.SquimObjective`.
         """
         model = squim_subjective_base()
-        model.load_state_dict(self._get_state_dict(dl_kwargs))
+        path = torchaudio.utils.download_asset(f"models/{self._path}")
+        state_dict = torch.load(path, weights_only=True)
+        model.load_state_dict(state_dict)
         model.eval()
         return model
 

torchaudio/version.py

@@ -1,2 +1,2 @@
-__version__ = '2.4.1'
-git_version = 'e8cbe17769796ce963fbc71b8990f1474774e6d2'
+__version__ = '2.5.1'
+git_version = '1661daf10599ca8889f092ec37814fabbe202bb0'

{torchaudio-2.4.1.dist-info → torchaudio-2.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: torchaudio
-Version: 2.4.1
+Version: 2.5.1
 Summary: An audio package for PyTorch
 Home-page: https://github.com/pytorch/audio
 Author: Soumith Chintala, David Pollack, Sean Naren, Peter Goldsborough, Moto Hira, Caroline Chen, Jeff Hwang, Zhaoheng Ni, Xiaohui Zhang
@@ -24,7 +24,7 @@ Classifier: Topic :: Multimedia :: Sound/Audio
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: torch (==2.4.1)
+Requires-Dist: torch (==2.5.1)
 
 torchaudio: an audio library for PyTorch
 ========================================

{torchaudio-2.4.1.dist-info → torchaudio-2.5.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 torchaudio/__init__.py,sha256=VSnZ6s4e5clAj7f7aCXBZt9amskeXg1j19txAQBQ2Iw,892
 torchaudio/kaldi_io.py,sha256=TwS2YgSLlJwOXjNNsHBuXyxhKeKKpptVHLBV7QYZCas,5073
-torchaudio/version.py,sha256=kRlH9btoEy65fUB4sXJAoSd0TZxTo4-oMYovrav62rc,79
+torchaudio/version.py,sha256=f4bXuRI4ZuPihUu1mg1Iysl1uDGTQK32QPtKHBc8_og,79
 torchaudio/_backend/__init__.py,sha256=6zMYGajHaaCXUE_U7HuGLp0fqcviYAjBZdFDI4E7C-0,1631
 torchaudio/_backend/backend.py,sha256=hSrfZcj5FMzx5ZpwubN-LLMvBFb7ENyw7HvT_6pVYVU,1565
 torchaudio/_backend/common.py,sha256=55Y0r0MsdW6gvTOT_Zy60UGFXc60DfdJ7uvycJKK3is,1783
@@ -53,10 +53,10 @@ torchaudio/io/__init__.py,sha256=8nd6s_xuBh5iVzIvQ-qNlforukZzuCx36DyvCmHK748,297
 torchaudio/io/_effector.py,sha256=APDrIU2biwFsSVmhrXjelmc4ndcmb0JL-H189Zp689g,11870
 torchaudio/io/_playback.py,sha256=70IxGrGPkI1h4rz8_04SFCGsbbGZkTiUdRhbPOMLLgQ,2321
 torchaudio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-torchaudio/lib/_torchaudio.so,sha256=W7VyG4XnNeSUUdpTs-jDP1HfdIxbIYrTYKpSGkti3AA,167920
-torchaudio/lib/_torchaudio_sox.so,sha256=tH5YbiCCc3VpWFPh4XkRwiHxCh4D8i7075HGgD2JymQ,269488
-torchaudio/lib/libtorchaudio.so,sha256=N4eZ35PBgZhZ7UzoxqNrCK8c3r_tFGHw2IDpxcSJUC8,642136
-torchaudio/lib/libtorchaudio_sox.so,sha256=leDLgPh2epbc4GF9y74jzK1xPSsQnwolnMd75qH0xrw,160536
+torchaudio/lib/_torchaudio.so,sha256=BPwBXJVNRaDHOznXHXinPmnA_Cq5XTJv-v-o2M6k_oY,168480
+torchaudio/lib/_torchaudio_sox.so,sha256=JVtXTqHew37YYRE8CJjDkyQWb5MpXXTRzhklpRdJVPU,270056
+torchaudio/lib/libtorchaudio.so,sha256=AQSiaKeHyAaFo4NOm3_hTqs0bcvAUS0RS_g_uqL3b68,643624
+torchaudio/lib/libtorchaudio_sox.so,sha256=p7R8P-A_w_6zBA5QzUm5rSccOVN0u4dZHqJzvkbnuGg,160968
 torchaudio/models/__init__.py,sha256=BNMNGuwpJAFRsdtwHYQ6slGClkrUTu31_7mXh7FjeV4,1995
 torchaudio/models/_hdemucs.py,sha256=VPnQ73lA9lfAxRjZ85NCGJYP36mPNwTjS-TU4qelu_k,38242
 torchaudio/models/conformer.py,sha256=5IceU-jcZKofkHTTqRKoytubQ75MzZPrPlfkLsIlxeA,10068
@@ -83,7 +83,7 @@ torchaudio/models/wav2vec2/utils/import_fairseq.py,sha256=oCwG6qpG0bCXue2V56fjDc
 torchaudio/models/wav2vec2/utils/import_huggingface.py,sha256=1nVCipp-lOUAyl_-P103DWLUeTOZi9X_ffX93bOXxEk,5946
 torchaudio/pipelines/__init__.py,sha256=Xy8NmInKwTcNBHwLTTjHjrfczRLuQq8a67ENt1OTVXM,2745
 torchaudio/pipelines/_source_separation_pipeline.py,sha256=WxngB1d13H5IVqs4RPqSL53ZGYsJ3tnfCpxgc5FNSOM,4224
-torchaudio/pipelines/_squim_pipeline.py,sha256=JROOpywuRFl6JnWVMhgqM2eu3REKeXj-_Te5QJzOwoA,7170
+torchaudio/pipelines/_squim_pipeline.py,sha256=eCimTeoqNX8LilIQNGmb8UaRtnLIXa4LNShXFjodcZM,6280
 torchaudio/pipelines/rnnt_pipeline.py,sha256=Qy37z7v6d1jLOHd67zbRu21dgL6Fml1rTd7j4Jl1NsM,13749
 torchaudio/pipelines/_tts/__init__.py,sha256=PP7l8XzVURqelwuMJFgfOCv4fvzZunDiy90ZQlRkv7g,426
 torchaudio/pipelines/_tts/impl.py,sha256=Tig4_5sITJADwxN5eZGek7Ath_-e3sV8CTM5t6UpeUU,15374
@@ -131,16 +131,16 @@ torio/io/__init__.py,sha256=xz7REkkyfRhAASzVCAfoNruFtAGIx1I--usPAa2tMww,226
 torio/io/_streaming_media_decoder.py,sha256=vSylEWAB_JXOW-0E1n0zDM3Q3Vf1jc1-CNpdUSs13XU,34376
 torio/io/_streaming_media_encoder.py,sha256=rSTYmHdi7RPJ6YPgAyGJhbQvn4mcxLem3nlnr_ophTs,19722
 torio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-torio/lib/_torio_ffmpeg4.so,sha256=b_TdZzj-AezUQCBuJ3cK6sRCrqWOsy1UzzFCtaGebMs,485296
-torio/lib/_torio_ffmpeg5.so,sha256=7W-jJSyZUgEz-kDwcUxHNGAO48ourvhYpDWYWc-6tfI,485296
-torio/lib/_torio_ffmpeg6.so,sha256=T3JYwyfvj318c5nmj8a65ibJQbUnqEchxQBh_yL3oA8,485296
-torio/lib/libtorio_ffmpeg4.so,sha256=RgBAlORivWxGf8k7YCsXTMhLQ5qLK1nvuJdgebtLzgc,598952
-torio/lib/libtorio_ffmpeg5.so,sha256=KypfnzuxS6djUUM2Kt6PFpdb8sdf_PBnn3SB5TDuUIA,598952
-torio/lib/libtorio_ffmpeg6.so,sha256=U6eYmf019-jeyXxsdIVWyV6UhDEKY0cHd_ewtCWKNAg,598952
+torio/lib/_torio_ffmpeg4.so,sha256=7Y2wcq_r93u03gQAJGe8ykAy0n8jmy8QOIIf03iAyvQ,486272
+torio/lib/_torio_ffmpeg5.so,sha256=MdMrSQVkINUO_-jKjAX3EjhPrA2OViHlhK5HJGqe06Y,486272
+torio/lib/_torio_ffmpeg6.so,sha256=pNnerr3FDQuqn13qzjci0M1nuu11CmwFSHIoFi1FsYc,486272
+torio/lib/libtorio_ffmpeg4.so,sha256=2eIJ4XrK7e2la_9dIVOkwFYXyVPCZ1yafbgH1AtGwhQ,600216
+torio/lib/libtorio_ffmpeg5.so,sha256=CRUyLS5DxSgr5npV5iQ_l-VFn-fCtIHbJN8G2LHxGLc,600216
+torio/lib/libtorio_ffmpeg6.so,sha256=uM4W8JmxgLFexMdzfASi-3Ja1MUMcI5naic-SUF215A,600216
 torio/utils/__init__.py,sha256=ScHtnontymRDNn9qEIC0neue5mfG82yhB8bwETOb0Z4,56
 torio/utils/ffmpeg_utils.py,sha256=JsP2ptjQAE4U7Z_CSauQKH_k72wdu6nrBMfNHl9pIXQ,8026
-torchaudio-2.4.1.dist-info/LICENSE,sha256=k6WIYahYzBCOa2uDPgjnbosqZjOeSoAHyKWowf-cQNY,1338
-torchaudio-2.4.1.dist-info/METADATA,sha256=MxKylZWzis1wYS46kRTXVOPObA_AakU-qfUp9X7JC24,6355
-torchaudio-2.4.1.dist-info/WHEEL,sha256=TxhIs8gTYHxwp4FCX0Qsm-R4pefOsFmo9fAF25K_2Bc,105
-torchaudio-2.4.1.dist-info/top_level.txt,sha256=GT0MktEbHKoLnvd-6ii7_dhJVvshupOujk840BcHU4U,17
-torchaudio-2.4.1.dist-info/RECORD,,
+torchaudio-2.5.1.dist-info/LICENSE,sha256=k6WIYahYzBCOa2uDPgjnbosqZjOeSoAHyKWowf-cQNY,1338
+torchaudio-2.5.1.dist-info/METADATA,sha256=RcyC-WrPDB8jGnGByVu2pMbcYh2Yu7gjeBIl9JoibDA,6355
+torchaudio-2.5.1.dist-info/WHEEL,sha256=Tm_QejvISbl3Z8WVUXK95VHAPH8B9jvALhdipHJZaMw,105
+torchaudio-2.5.1.dist-info/top_level.txt,sha256=GT0MktEbHKoLnvd-6ii7_dhJVvshupOujk840BcHU4U,17
+torchaudio-2.5.1.dist-info/RECORD,,

{torchaudio-2.4.1.dist-info → torchaudio-2.5.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (72.2.0)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: false
 Tag: cp312-cp312-linux_aarch64