torchaudio 2.4.1__cp311-cp311-macosx_11_0_arm64.whl → 2.5.1__cp311-cp311-macosx_11_0_arm64.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

@@ -4,16 +4,21 @@ torio/io/__init__.py,sha256=xz7REkkyfRhAASzVCAfoNruFtAGIx1I--usPAa2tMww,226
 torio/io/_streaming_media_encoder.py,sha256=rSTYmHdi7RPJ6YPgAyGJhbQvn4mcxLem3nlnr_ophTs,19722
 torio/utils/__init__.py,sha256=ScHtnontymRDNn9qEIC0neue5mfG82yhB8bwETOb0Z4,56
 torio/utils/ffmpeg_utils.py,sha256=JsP2ptjQAE4U7Z_CSauQKH_k72wdu6nrBMfNHl9pIXQ,8026
-torio/lib/libtorio_ffmpeg5.so,sha256=sQo9MM02d3JkDIo8_lvYcmRXz4M82l3fcorr3bYFakU,515360
-torio/lib/_torio_ffmpeg4.so,sha256=ALkhcHU_9Qrnsux5Am8NFC6hUJRLb9byzne5gfn9Rpk,518960
+torio/lib/libtorio_ffmpeg5.so,sha256=v2d9zWUMPAap-bewDKTOtr3qwZq_QWdVRIpBXyJ-WHs,515616
+torio/lib/_torio_ffmpeg4.so,sha256=yQZo00V1hOYyJWpI6QlQLEpe6vxp-_b4bp_A_Xw9AOA,519536
 torio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-torio/lib/_torio_ffmpeg5.so,sha256=G1O6EEXFeGGLaaNiG0rNTG2oLjcEgjw_S_PshZ6DJJk,518960
-torio/lib/libtorio_ffmpeg4.so,sha256=MrhtCGCG1dLNOiALR3A5vuj9x7Kau0HWzrLupOrF6t0,515360
-torio/lib/_torio_ffmpeg6.so,sha256=3y15YoJY0q0_BttMt9y39SLzVlx5ZGPiy0x7obU72-k,518960
-torio/lib/libtorio_ffmpeg6.so,sha256=g3Tad3Que-NiLG-GNblsddyvVeR1vH4XLaLaHfWZV4s,515360
+torio/lib/_torio_ffmpeg5.so,sha256=0UkNU3r9NgsBH3Etmb-uMK1uEroZXVeQWyEkTfAXb1w,519536
+torio/lib/libtorio_ffmpeg4.so,sha256=KF0U6Dhee7L-iEZ02hOjiQw84v1-pvb7IAzWoiBJTxw,515616
+torio/lib/_torio_ffmpeg6.so,sha256=AnMg7F0I2kg0BAyIeG9p7SDMgQ6-CkOHwdsVAQ5auag,519536
+torio/lib/libtorio_ffmpeg6.so,sha256=xRG20BAsRZwggzScIS7yQEQAUF-emTo3iVu3JiUoNrg,515616
 torio/_extension/__init__.py,sha256=q5jjeOhSrzqn0WTEwrx61Fr13aCjb7IQCDGsBqAdGEU,313
 torio/_extension/utils.py,sha256=ktE0L_z-RF1qkpLVGgdG4DEGHa2Zn6uokOAmwC7Evvo,4904
-torchaudio/version.py,sha256=kRlH9btoEy65fUB4sXJAoSd0TZxTo4-oMYovrav62rc,79
+torchaudio-2.5.1.dist-info/RECORD,,
+torchaudio-2.5.1.dist-info/LICENSE,sha256=k6WIYahYzBCOa2uDPgjnbosqZjOeSoAHyKWowf-cQNY,1338
+torchaudio-2.5.1.dist-info/WHEEL,sha256=PF0Mgwtv_oDs_mWYUtLpGhYaIa8jOhCqBxSqdxy2gqE,109
+torchaudio-2.5.1.dist-info/top_level.txt,sha256=GT0MktEbHKoLnvd-6ii7_dhJVvshupOujk840BcHU4U,17
+torchaudio-2.5.1.dist-info/METADATA,sha256=RcyC-WrPDB8jGnGByVu2pMbcYh2Yu7gjeBIl9JoibDA,6355
+torchaudio/version.py,sha256=f4bXuRI4ZuPihUu1mg1Iysl1uDGTQK32QPtKHBc8_og,79
 torchaudio/kaldi_io.py,sha256=TwS2YgSLlJwOXjNNsHBuXyxhKeKKpptVHLBV7QYZCas,5073
 torchaudio/__init__.py,sha256=VSnZ6s4e5clAj7f7aCXBZt9amskeXg1j19txAQBQ2Iw,892
 torchaudio/_internal/__init__.py,sha256=gjU8g9HhVd9hHrHXJM0xOlZL6cT8ktO60MN8RHI6ZbA,241
@@ -47,7 +52,7 @@ torchaudio/io/__init__.py,sha256=8nd6s_xuBh5iVzIvQ-qNlforukZzuCx36DyvCmHK748,297
 torchaudio/pipelines/_source_separation_pipeline.py,sha256=WxngB1d13H5IVqs4RPqSL53ZGYsJ3tnfCpxgc5FNSOM,4224
 torchaudio/pipelines/__init__.py,sha256=Xy8NmInKwTcNBHwLTTjHjrfczRLuQq8a67ENt1OTVXM,2745
 torchaudio/pipelines/rnnt_pipeline.py,sha256=Qy37z7v6d1jLOHd67zbRu21dgL6Fml1rTd7j4Jl1NsM,13749
-torchaudio/pipelines/_squim_pipeline.py,sha256=JROOpywuRFl6JnWVMhgqM2eu3REKeXj-_Te5QJzOwoA,7170
+torchaudio/pipelines/_squim_pipeline.py,sha256=eCimTeoqNX8LilIQNGmb8UaRtnLIXa4LNShXFjodcZM,6280
 torchaudio/pipelines/_wav2vec2/aligner.py,sha256=pIWRgQ-kdYUxtL8bdc0qk9wBjwRrHY1uSWL3L4e2vxs,2709
 torchaudio/pipelines/_wav2vec2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 torchaudio/pipelines/_wav2vec2/utils.py,sha256=Q8_fWOR2JDnHu0TTRmHzRjI3BOJa0hGIAl0cjtALgsQ,6971
@@ -127,11 +132,11 @@ torchaudio/_backend/soundfile_backend.py,sha256=qJHEEXU1egCkPJ2Y9uJWFvVhW3AqDZ7z
 torchaudio/transforms/__init__.py,sha256=Tp1o4haiJAV3MRJenmvGXFbmt-RE4qM_pd6U3Ghohqw,1270
 torchaudio/transforms/_transforms.py,sha256=QHrEsxxxm1bPd5dltPeTcNOsMBu0Ecxa2oe6GIX-nvk,86872
 torchaudio/transforms/_multi_channel.py,sha256=GZ2rrwFt2KtSG7At7kS9Bqh1KmYYw0HwcUnEjc-AWr8,22221
-torchaudio/lib/libtorchaudio.so,sha256=BfZd0LNvxxqJuUJLix_1DtBXJ-KSQ7J4Kz9IufeWl0M,701808
+torchaudio/lib/libtorchaudio.so,sha256=Psdh1yyGz5AhafYLcCz1Hk5hX9wYHm8GZcBKbcV_J1g,721808
 torchaudio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-torchaudio/lib/_torchaudio.so,sha256=lWu3ZLJOuJN6MO63tGxVmirVOVWFtx2d2kkjHClUfCM,190976
-torchaudio/lib/libtorchaudio_sox.so,sha256=z9qrp0Wfzey8PW0Ger49BxNecTJPejHhpJkkp4LyqM0,173088
-torchaudio/lib/_torchaudio_sox.so,sha256=skO_HaoqN4HrQ4V4fhX3FxGiT8We4du7huBOa98VN5I,318768
+torchaudio/lib/_torchaudio.so,sha256=rpoqlR_vlU82Cki67DWxa-yQ6gLiwY3BM-21NE05nTQ,191152
+torchaudio/lib/libtorchaudio_sox.so,sha256=Y3pvXBjcVhGXAHxPyD0q8QLSkhKMu3qmzoX5orF5n2s,173280
+torchaudio/lib/_torchaudio_sox.so,sha256=tyiGXfL3KhX7IEsDTYZZ1topbtootKS0tU315jaqYf8,318944
 torchaudio/_extension/__init__.py,sha256=lQPB8K7VSxWmtTEiMmF-u7WVq1O10_t5nEghkjCf4Ks,2202
 torchaudio/_extension/utils.py,sha256=4FTD6xwcSLqVJ3Kmpx5cvJp1oAUKmWwRjwuxpcbrmzw,6258
 torchaudio/sox_effects/sox_effects.py,sha256=7cHpPFRJ_pZuohHMnX9JIhiVmIJGYntSmgT6QH5GNMA,10981
@@ -140,8 +145,3 @@ torchaudio/functional/filtering.py,sha256=5SAPz-0sioU5EYX4VqOnbPAys8jtPSpFnpzbVn
 torchaudio/functional/__init__.py,sha256=l-gC2WyY5COabU0lhkUS8EnwOYdEYR_6234OyoAIgnU,2357
 torchaudio/functional/_alignment.py,sha256=wmDeohWvuoYORYDeIRxnYUhUqv1uCUkaCZYLEK_ryUg,4695
 torchaudio/functional/functional.py,sha256=c8qr3mmPXLi40N4NCLcpHvQeUIuoNtbszksjtruC15g,96006
-torchaudio-2.4.1.dist-info/RECORD,,
-torchaudio-2.4.1.dist-info/LICENSE,sha256=k6WIYahYzBCOa2uDPgjnbosqZjOeSoAHyKWowf-cQNY,1338
-torchaudio-2.4.1.dist-info/WHEEL,sha256=l8pL05u3PzAsBDEfGhzraFoAjYvGQLMHHDRUD7RvxIM,109
-torchaudio-2.4.1.dist-info/top_level.txt,sha256=GT0MktEbHKoLnvd-6ii7_dhJVvshupOujk840BcHU4U,17
-torchaudio-2.4.1.dist-info/METADATA,sha256=MxKylZWzis1wYS46kRTXVOPObA_AakU-qfUp9X7JC24,6355

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

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (72.1.0)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: false
 Tag: cp311-cp311-macosx_11_0_arm64