torchaudio 2.4.0__cp39-cp39-manylinux2014_aarch64.whl → 2.5.0__cp39-cp39-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.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,6 +1,6 @@
 torchaudio/__init__.py,sha256=VSnZ6s4e5clAj7f7aCXBZt9amskeXg1j19txAQBQ2Iw,892
 torchaudio/kaldi_io.py,sha256=TwS2YgSLlJwOXjNNsHBuXyxhKeKKpptVHLBV7QYZCas,5073
-torchaudio/version.py,sha256=JUO1bPDj-WIPcr-DwAEpP5gRiwdtsHyJViOL86zXNxQ,79
+torchaudio/version.py,sha256=OAqErR7sj83_w1z5wQ20r9YwFkll8NQIGB8v8_ZJuGg,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,12 @@ 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=F-vJd-7y6R7gq-MYyHjUkWkQ5ekVwHsSlBZ-JtVqDMc,167736
-torchaudio/lib/_torchaudio_sox.so,sha256=HwkfdtN5rYqBTFp7hUyI0-U-6IlCAPZ-FLqzTlaEJB8,269272
-torchaudio/lib/libtorchaudio.so,sha256=CzVYk--0dEr4x85128cexB026r0rt4WHWtAL9s8TD7I,642136
-torchaudio/lib/libtorchaudio_sox.so,sha256=PGWnV_sfn_DwyfJRlG8IfoTn474N0JZ45P9Vpa2M_O8,160536
+torchaudio/lib/_torchaudio.so,sha256=A149aZkNE6m_DAjXNLgUM662d4xp15tPEzOncB5xTOQ,169664
+torchaudio/lib/_torchaudio_sox.so,sha256=HGMXc2eo1Lq99tig3_-HHdOSL0_fgAXCzFmp6AaGh6I,277720
+torchaudio/lib/libctc_prefix_decoder.so,sha256=vU_ivW6VsRGfFx5tpqW3QkvtyHRxMRvNBftgDUAPq78,4872120
+torchaudio/lib/libtorchaudio.so,sha256=1U9K0X-bFTTDW6uUK-8kFe2-hlgJXvr6Z2opNY3XyHM,2374416
+torchaudio/lib/libtorchaudio_sox.so,sha256=auURi4-pordIt43F7LE9rzNSTuOobhKs6717t0vTSrg,162536
+torchaudio/lib/pybind11_prefixctc.so,sha256=3jxWNFC2mMXv3SxqXv3d-U3VP3UIo852x0donEROLjU,172360
 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 +85,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 +133,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=HxV3HVEDXTXjFEy0flGBlFfRjDu7m2d5q20Jpd-p1v4,485016
-torio/lib/_torio_ffmpeg5.so,sha256=vTXqUWsc689A1nhPY5Q9NsO0a6s8QuX-3vYAz_vW8Vw,485016
-torio/lib/_torio_ffmpeg6.so,sha256=yTO5c_BNYJ2FrxYUU42mDhbGDPNK84s70kiuWsiMeQE,485016
-torio/lib/libtorio_ffmpeg4.so,sha256=wGRLsRsNqGmWkbHDNlH4gM9r6lxOvxbM2AtU3g-S0Hg,598952
-torio/lib/libtorio_ffmpeg5.so,sha256=d0vYdgLAYMfwAoRos1FIT9cYI4MMccdmTsfG6_vr1m4,598952
-torio/lib/libtorio_ffmpeg6.so,sha256=wPLnoTtgkL3EqAuiVzaNsrRdKx1FyKN8C7QBjlYSkzA,598952
+torio/lib/_torio_ffmpeg4.so,sha256=NjuYycF9mGr2Cz834tI5W0HEHNKYW5GTbTo4HYjqr3I,498120
+torio/lib/_torio_ffmpeg5.so,sha256=ebll9GNU2-0kBfIRDI-_dSVLKGfBc-6OO_YLujGwKsU,498120
+torio/lib/_torio_ffmpeg6.so,sha256=UB9MxCT4uHeBB0L2aBXkTljXzUJZTLbxBn3ivzoK0fs,498120
+torio/lib/libtorio_ffmpeg4.so,sha256=EN8LWoF-n7Mw-4-3Nqm0Gm3wV8P4VGo9TDfrw0X_j2o,621664
+torio/lib/libtorio_ffmpeg5.so,sha256=pQR6wsAYhuFU5Y6NRTZ3EfjOW3FCej2bddZL_mq-jr4,621664
+torio/lib/libtorio_ffmpeg6.so,sha256=P0brZxuxacnxTCM9kNSV-EOtwvxpEXhjWGd1suqMwSs,621664
 torio/utils/__init__.py,sha256=ScHtnontymRDNn9qEIC0neue5mfG82yhB8bwETOb0Z4,56
 torio/utils/ffmpeg_utils.py,sha256=JsP2ptjQAE4U7Z_CSauQKH_k72wdu6nrBMfNHl9pIXQ,8026
-torchaudio-2.4.0.dist-info/LICENSE,sha256=k6WIYahYzBCOa2uDPgjnbosqZjOeSoAHyKWowf-cQNY,1338
-torchaudio-2.4.0.dist-info/METADATA,sha256=h_ca-IgoHbW5vyqG0nZ8A_wal0rB3TXj7M-uOkagteU,6349
-torchaudio-2.4.0.dist-info/WHEEL,sha256=SiTB-TT6RTfH2k1CR4Re6_8was4k1KvPQ3fTSNKoczw,103
-torchaudio-2.4.0.dist-info/top_level.txt,sha256=GT0MktEbHKoLnvd-6ii7_dhJVvshupOujk840BcHU4U,17
-torchaudio-2.4.0.dist-info/RECORD,,
+torchaudio-2.5.0.dist-info/LICENSE,sha256=k6WIYahYzBCOa2uDPgjnbosqZjOeSoAHyKWowf-cQNY,1338
+torchaudio-2.5.0.dist-info/METADATA,sha256=i3S98TlBHgnw49cW7QAEuT2KO7X172UwXprekeOfE-4,6355
+torchaudio-2.5.0.dist-info/WHEEL,sha256=KOV3ZyHTxJBAHqM-DFZZjfeqa6iJsa5EhllyY87m6WA,103
+torchaudio-2.5.0.dist-info/top_level.txt,sha256=GT0MktEbHKoLnvd-6ii7_dhJVvshupOujk840BcHU4U,17
+torchaudio-2.5.0.dist-info/RECORD,,

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

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