torchaudio 2.4.0__cp311-cp311-macosx_11_0_arm64.whl → 2.5.0__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.0'
-git_version = '69d40773dc4ed86643820c21a8a880e4d074a46e'
+__version__ = '2.5.0'
+git_version = '56bc006d56a0d4960de6a1e0b6340cba4eda05cd'

{torchaudio-2.4.0.dist-info → torchaudio-2.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: torchaudio
-Version: 2.4.0
+Version: 2.5.0
 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.0)
+Requires-Dist: torch (==2.5.0)
 
 torchaudio: an audio library for PyTorch
 ========================================
@@ -79,7 +79,7 @@ If you find this package useful, please cite as:
 ```bibtex
 @article{yang2021torchaudio,
   title={TorchAudio: Building Blocks for Audio and Speech Processing},
-  author={Yao-Yuan Yang and Moto Hira and Zhaoheng Ni and Anjali Chourdia and Artyom Astafurov and Caroline Chen and Ching-Feng Yeh and Christian Puhrsch and David Pollack and Dmitriy Genzel and Donny Greenberg and Edward Z. Yang and Jason Lian and Jay Mahadeokar and Jeff Hwang and Ji Chen and Peter Goldsborough and Prabhat Roy and Sean Narenthiran and Shinji Watanabe and Soumith Chintala and Vincent Quenneville-Bélair and Yangyang Shi},
+  author={Yao-Yuan Yang and Moto Hira and Zhaoheng Ni and Anjali Chourdia and Artyom Astafurov and Caroline Chen and Ching-Feng Yeh and Christian Puhrsch and David Pollack and Dmitriy Genzel and Donny Greenberg and Edward Z. Yang and Jason Lian and Jay Mahadeokar and Jeff Hwang and Ji Chen and Peter Goldsborough and Prabhat Roy and Sean Narenthiran and Shinji Watanabe and Soumith Chintala and Vincent Quenneville-B��lair and Yangyang Shi},
   journal={arXiv preprint arXiv:2110.15018},
   year={2021}
 }

{torchaudio-2.4.0.dist-info → torchaudio-2.5.0.dist-info}/RECORD

@@ -1,24 +1,24 @@
+torchaudio-2.5.0.dist-info/RECORD,,
+torchaudio-2.5.0.dist-info/LICENSE,sha256=k6WIYahYzBCOa2uDPgjnbosqZjOeSoAHyKWowf-cQNY,1338
+torchaudio-2.5.0.dist-info/WHEEL,sha256=PF0Mgwtv_oDs_mWYUtLpGhYaIa8jOhCqBxSqdxy2gqE,109
+torchaudio-2.5.0.dist-info/top_level.txt,sha256=GT0MktEbHKoLnvd-6ii7_dhJVvshupOujk840BcHU4U,17
+torchaudio-2.5.0.dist-info/METADATA,sha256=i3S98TlBHgnw49cW7QAEuT2KO7X172UwXprekeOfE-4,6355
 torio/__init__.py,sha256=aX9s0XAHxHhEXE1akQt74BZ0cMUDgBPhaYHQH1lCbXQ,111
 torio/io/_streaming_media_decoder.py,sha256=vSylEWAB_JXOW-0E1n0zDM3Q3Vf1jc1-CNpdUSs13XU,34376
 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=C48UU5N7un-YMsvdArilgz7DoPc-Jw9ldPmZYg2Z-yU,515360
-torio/lib/_torio_ffmpeg4.so,sha256=fcf1jIs5w0xR_FFQ2GVXSiekXVnrQowsfq6MpnZo6Qg,518960
+torio/lib/libtorio_ffmpeg5.so,sha256=2vAf0gIyIhkqq9BOEGFfk41_GSscvTr2Jjhj51Oo5Os,515616
+torio/lib/_torio_ffmpeg4.so,sha256=-T0-WVlTqWBLcgrRthKEGzDbJurPs9T8iLJplH_XNB4,519536
 torio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-torio/lib/_torio_ffmpeg5.so,sha256=r11G8HZ8D_yKDwosvM9-Gxs_ggkK7z0bgKv1A90wjZo,518960
-torio/lib/libtorio_ffmpeg4.so,sha256=YCbp2rgwjj7TaOj0BV-x1tzFVABOPPcV4om3KlCN6Sc,515360
-torio/lib/_torio_ffmpeg6.so,sha256=cQwuDjATQiWdnVOFy1Vd_EJw5WNu8pKh2-4YLnAjRU8,518960
-torio/lib/libtorio_ffmpeg6.so,sha256=-ZarKv2MOKsuzfkNRoux30sfGfiToPdr_2SMaex3QFM,515360
+torio/lib/_torio_ffmpeg5.so,sha256=H1N-XaAwmXoLqNckRXFZNVU0SKmWMfkUQUxVE7WS_Zo,519536
+torio/lib/libtorio_ffmpeg4.so,sha256=cHKUY2T-DbOnWVAP02-Cx-JntWKeb7rNSFpFeXBeoGA,515616
+torio/lib/_torio_ffmpeg6.so,sha256=Hw1JRDBjqquRSFbuQHVAYd_OvA62x2gvkC5vmBVSEVw,519536
+torio/lib/libtorio_ffmpeg6.so,sha256=RDM1sxxXvUs9-DV23BcTBwNkqRIm3YPR1h6j6XID-lg,515616
 torio/_extension/__init__.py,sha256=q5jjeOhSrzqn0WTEwrx61Fr13aCjb7IQCDGsBqAdGEU,313
 torio/_extension/utils.py,sha256=ktE0L_z-RF1qkpLVGgdG4DEGHa2Zn6uokOAmwC7Evvo,4904
-torchaudio-2.4.0.dist-info/RECORD,,
-torchaudio-2.4.0.dist-info/LICENSE,sha256=k6WIYahYzBCOa2uDPgjnbosqZjOeSoAHyKWowf-cQNY,1338
-torchaudio-2.4.0.dist-info/WHEEL,sha256=fHvRBzKCrQbGWlIL1eIB4x2YIBk9sCn_QC5-FS8Vm74,110
-torchaudio-2.4.0.dist-info/top_level.txt,sha256=GT0MktEbHKoLnvd-6ii7_dhJVvshupOujk840BcHU4U,17
-torchaudio-2.4.0.dist-info/METADATA,sha256=MRLtyEJp6m5kudj5YIQFZvfBu-i7rZvloNv56gXFljM,6351
-torchaudio/version.py,sha256=JUO1bPDj-WIPcr-DwAEpP5gRiwdtsHyJViOL86zXNxQ,79
+torchaudio/version.py,sha256=OAqErR7sj83_w1z5wQ20r9YwFkll8NQIGB8v8_ZJuGg,79
 torchaudio/kaldi_io.py,sha256=TwS2YgSLlJwOXjNNsHBuXyxhKeKKpptVHLBV7QYZCas,5073
 torchaudio/__init__.py,sha256=VSnZ6s4e5clAj7f7aCXBZt9amskeXg1j19txAQBQ2Iw,892
 torchaudio/_internal/__init__.py,sha256=gjU8g9HhVd9hHrHXJM0xOlZL6cT8ktO60MN8RHI6ZbA,241
@@ -52,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
@@ -132,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=Lhne3nJZ9h9KarDIZHbVmkD1FWEH_WpwdwS5DR5wLmM,701808
+torchaudio/lib/libtorchaudio.so,sha256=GpBjhBU_AfYN-WJ8SBkSEN1E31Pvv4V10V5v0LbNZWs,721808
 torchaudio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-torchaudio/lib/_torchaudio.so,sha256=l9LIn07BDRZ2iCuA3nqWr7fcRzAtseD8Axl1_hpJYNE,190976
-torchaudio/lib/libtorchaudio_sox.so,sha256=s04TYpM1BfODnerLr0N_i5N_221NtsCLU5VwVvwO2sg,173088
-torchaudio/lib/_torchaudio_sox.so,sha256=yV6wKjdFHg8J9oa_8ipfMAQLUjxck6ie5B_8x86oBF8,318768
+torchaudio/lib/_torchaudio.so,sha256=a8luSeJmnnSeDoQnzutRe3hpQfh5MdgxyEfLgnrk8SE,191152
+torchaudio/lib/libtorchaudio_sox.so,sha256=aHmXSk250OdSn1YlpEvxLpPW8JhTgxNTogaczheD9nE,173280
+torchaudio/lib/_torchaudio_sox.so,sha256=-hJ4LzvWS4aPxqkzsuilJcBs7iJxfYE-1yhkqhPlnSM,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

{torchaudio-2.4.0.dist-info → torchaudio-2.5.0.dist-info}/WHEEL

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