torchaudio 2.4.1__cp312-cp312-macosx_11_0_arm64.whl → 2.5.1__cp312-cp312-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=0E1of7um42pomjAoaM3pQLWHEb32MyVUT9_N3O-t0DU,515360
-torio/lib/_torio_ffmpeg4.so,sha256=UzHWyVg5RmN2Sjr0R0tDJCnwdnoFKaMORWfddK0Gqdw,520880
+torio/lib/libtorio_ffmpeg5.so,sha256=NHKLC__a2P2Gn-ffqPfKqZTIeNL_9z3EcbNPpRNl4SQ,515616
+torio/lib/_torio_ffmpeg4.so,sha256=NlhSBJZoBj2d_gY6xx-apwbD6xmlBj4ED0i4XIafBBk,521824
 torio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-torio/lib/_torio_ffmpeg5.so,sha256=twZ6kyQbXbVCZx0wV7HHm9GrLSMsQEiGULQnENYxiLA,520880
-torio/lib/libtorio_ffmpeg4.so,sha256=frPi4nOqTkqpW3MyJw94jJlq7Iox079h_fbHYFidiuQ,515360
-torio/lib/_torio_ffmpeg6.so,sha256=r86icAeVPSU6OG3oOlvRMspi9HLdOchQR4x1Hg75FI8,520880
-torio/lib/libtorio_ffmpeg6.so,sha256=F-nyfNaz8S5jA9rTo_Ofz7ath31m4-Y94Ypt0jns7P0,515360
+torio/lib/_torio_ffmpeg5.so,sha256=MZPDcyOhw8xbEcNkReIPkPSmahLF5iTlGxc8V_R8zQo,521824
+torio/lib/libtorio_ffmpeg4.so,sha256=ToOf-xxGR-cKuyZpEbk4Rs38tt0Z1dUkeuZWlCiU35s,515616
+torio/lib/_torio_ffmpeg6.so,sha256=N8bW5RfZ3H1DfeuckSczU-RR0CMDY4pylFS9vDyKmJA,521824
+torio/lib/libtorio_ffmpeg6.so,sha256=eiP8cfxCErkHuyTzjjPnKc-axlYKQ6wAzFGpSVoQioE,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=lWey-nzGdePMz7fsIJ1fIMVKiFg5IkOcUVqkuIpdbws,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=pUsE8SdoOESITEVt4Zf0FK16wiwm7dlmG-2f5rgH-xQ,701808
+torchaudio/lib/libtorchaudio.so,sha256=G0ZlBFZU3ch4cBNaAZSFFBYXimsis-moAfdy5FXWrDY,721808
 torchaudio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-torchaudio/lib/_torchaudio.so,sha256=mkksKoESIv8gNKWbSNp_02YxETGJvOfJDa2T_PKpJAE,191552
-torchaudio/lib/libtorchaudio_sox.so,sha256=G5wBS2aTgFgMzfBKQfgJgok6zFypXKLA8K8fg0r_3rI,173088
-torchaudio/lib/_torchaudio_sox.so,sha256=ZOH-8-X47AA4vRc7RwfOxQhHyZ67a5EudkipiqgeWBU,319872
+torchaudio/lib/_torchaudio.so,sha256=p_32_7A-6JNkPv5x4L2dFqihGy7hTmBQ_nZLnTELXh4,192112
+torchaudio/lib/libtorchaudio_sox.so,sha256=GWLrqvkRX18HHwjCGjLXQdlVEwXf9-q3zjxVrZXHYtU,173280
+torchaudio/lib/_torchaudio_sox.so,sha256=0uqnu0iZwkmyHPyEp2NYBESfT2MMz-wxMeRymDv_j1Y,320416
 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=fDzG1t-TOTrzpRgXgdpro6xhK4j0kqVYadqkwQdD_0I,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: cp312-cp312-macosx_11_0_arm64