torchaudio 2.4.1__cp39-cp39-manylinux1_x86_64.whl → 2.5.1__cp39-cp39-manylinux1_x86_64.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+cu121'
-git_version = 'e8cbe17769796ce963fbc71b8990f1474774e6d2'
+__version__ = '2.5.1+cu124'
+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

@@ -38,7 +38,7 @@ torchaudio/pipelines/_tts/interface.py,sha256=yUaS0UK3PTRruYXRWFil7lAhr-1iYiyBaD
 torchaudio/pipelines/_tts/impl.py,sha256=Tig4_5sITJADwxN5eZGek7Ath_-e3sV8CTM5t6UpeUU,15374
 torchaudio/pipelines/_tts/__init__.py,sha256=PP7l8XzVURqelwuMJFgfOCv4fvzZunDiy90ZQlRkv7g,426
 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/_source_separation_pipeline.py,sha256=WxngB1d13H5IVqs4RPqSL53ZGYsJ3tnfCpxgc5FNSOM,4224
 torchaudio/pipelines/__init__.py,sha256=Xy8NmInKwTcNBHwLTTjHjrfczRLuQq8a67ENt1OTVXM,2745
 torchaudio/models/wav2vec2/utils/import_huggingface.py,sha256=1nVCipp-lOUAyl_-P103DWLUeTOZi9X_ffX93bOXxEk,5946
@@ -65,12 +65,12 @@ torchaudio/models/conv_tasnet.py,sha256=v-DI_Ej9FCBBbSH-Spkh3tzq8rkBhbQNA-Wp52Uf
 torchaudio/models/conformer.py,sha256=5IceU-jcZKofkHTTqRKoytubQ75MzZPrPlfkLsIlxeA,10068
 torchaudio/models/_hdemucs.py,sha256=VPnQ73lA9lfAxRjZ85NCGJYP36mPNwTjS-TU4qelu_k,38242
 torchaudio/models/__init__.py,sha256=BNMNGuwpJAFRsdtwHYQ6slGClkrUTu31_7mXh7FjeV4,1995
-torchaudio/lib/pybind11_prefixctc.so,sha256=IQFhxXnmiZ9rbX3ruBwXLGktd2UzL4Uy7_b-jhMJW3Y,213720
-torchaudio/lib/libtorchaudio_sox.so,sha256=mBfgffL1D0QuIc4_wGLmlkH_QhLwsrcJhHJt-My-UMU,151672
-torchaudio/lib/libtorchaudio.so,sha256=3jTGjVh17Z-bimhl3885v8QdYXtanX2H1deDOVgSlRQ,2385008
-torchaudio/lib/libctc_prefix_decoder.so,sha256=5CUjp2ze6xLvQFJTGdTFtf6-7V7noXJ0znRxHKrCWHM,4901512
-torchaudio/lib/_torchaudio_sox.so,sha256=PEcpWBOW3Ngq-t8EoUm4yqhe8LpbfyabV_-gfn20FtE,261528
-torchaudio/lib/_torchaudio.so,sha256=1f9yH8Eksd9zcX4iJmNR_jJHyb6XvkK5Y3Ly03kWsbc,133664
+torchaudio/lib/pybind11_prefixctc.so,sha256=Gld7FpERwFTIeuudycR4cdOfPJ4c27-a0Qw_KhNQp3o,214032
+torchaudio/lib/libtorchaudio_sox.so,sha256=NZa1K0VpEJQ2c4wCOEaKZBt8sXk61GmhqNevxUVgMDE,151800
+torchaudio/lib/libtorchaudio.so,sha256=u8mNLWmN3XslKAPGa2teyLoayhjNsR404qqSOdguoV0,2452168
+torchaudio/lib/libctc_prefix_decoder.so,sha256=gTk_g1YesJw33vALSbAEqnqpITU-CtaArlk3bAG7P9g,4943448
+torchaudio/lib/_torchaudio_sox.so,sha256=385LfAs86Q66NzkVyuMJJeK38qXjWCoUIaYnkWiUVZw,261848
+torchaudio/lib/_torchaudio.so,sha256=6DvUJ5aSN04akNpG5GDHGTicQ-HYBM0Hu1p2VOkZcmA,133976
 torchaudio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 torchaudio/io/_playback.py,sha256=70IxGrGPkI1h4rz8_04SFCGsbbGZkTiUdRhbPOMLLgQ,2321
 torchaudio/io/_effector.py,sha256=APDrIU2biwFsSVmhrXjelmc4ndcmb0JL-H189Zp689g,11870
@@ -123,22 +123,22 @@ torchaudio/_backend/ffmpeg.py,sha256=oL_whDjkPtHzo6HJLiEPlGHdrOqzjlu81g-vlaNkRBA
 torchaudio/_backend/common.py,sha256=55Y0r0MsdW6gvTOT_Zy60UGFXc60DfdJ7uvycJKK3is,1783
 torchaudio/_backend/backend.py,sha256=hSrfZcj5FMzx5ZpwubN-LLMvBFb7ENyw7HvT_6pVYVU,1565
 torchaudio/_backend/__init__.py,sha256=6zMYGajHaaCXUE_U7HuGLp0fqcviYAjBZdFDI4E7C-0,1631
-torchaudio/version.py,sha256=kHSQdJV4dbkRXSa8p8mmPVISNrJM_xbvj6KqrGocyoM,85
+torchaudio/version.py,sha256=Pz3LXTV6k3pr9rVL0ttY6CGXK1gGt0EnSjg3H0GpIcw,85
 torchaudio/kaldi_io.py,sha256=TwS2YgSLlJwOXjNNsHBuXyxhKeKKpptVHLBV7QYZCas,5073
 torchaudio/__init__.py,sha256=VSnZ6s4e5clAj7f7aCXBZt9amskeXg1j19txAQBQ2Iw,892
-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=afZrAwf6IoUB8kNXkHSgx85l0rlcsHjhnw3BOzWgzZI,102
-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=c7fm7CntAP4CB76e4qgkdpfoul4eIbgsFyo7c9yN4jE,102
+torchaudio-2.5.1.dist-info/top_level.txt,sha256=GT0MktEbHKoLnvd-6ii7_dhJVvshupOujk840BcHU4U,17
+torchaudio-2.5.1.dist-info/RECORD,,
 torio/utils/ffmpeg_utils.py,sha256=JsP2ptjQAE4U7Z_CSauQKH_k72wdu6nrBMfNHl9pIXQ,8026
 torio/utils/__init__.py,sha256=ScHtnontymRDNn9qEIC0neue5mfG82yhB8bwETOb0Z4,56
-torio/lib/libtorio_ffmpeg6.so,sha256=V6UV_KYFk7yMgWoZfmGZRXMKy7vaufsXwPbJbcwNkq8,647456
-torio/lib/libtorio_ffmpeg5.so,sha256=csY_YwyHHBeoI_JfWyY1o1go-b8M5RXfLGgIUlVlhZg,647456
-torio/lib/libtorio_ffmpeg4.so,sha256=6scqRA14m1gDmUivWAprOyyeY_H1eFlN0w4j7O1TJPI,647456
-torio/lib/_torio_ffmpeg6.so,sha256=UIZoNlACdxXFgfOTaRNxJ8mOt1iQQw5cY0jjepuMZF4,484000
-torio/lib/_torio_ffmpeg5.so,sha256=PZZQvJqUrH8A120xEkfSZu_CJUvZ7DafeFb57O3itxk,484000
-torio/lib/_torio_ffmpeg4.so,sha256=TX5FDfPOTFYLjwCI5vBTGl8SJEmplQyLLpEj5tbEBl0,484000
+torio/lib/libtorio_ffmpeg6.so,sha256=pIKn4TJmaryUJwnFXaof23qYetQ4b91Tj96p7RrEhFQ,648016
+torio/lib/libtorio_ffmpeg5.so,sha256=uLpkHxrvmH5s1_3aVYFHp6Zwn0Q07xorgRcSSsEXy4c,648016
+torio/lib/libtorio_ffmpeg4.so,sha256=IX7RRss465a0L6TOdxWlTNVo8EqSpwgNxLeep5Jzeqk,652112
+torio/lib/_torio_ffmpeg6.so,sha256=-z5x25SnevpJhiGkpqlIubCzI4qhMlm8I_t2oAuPZ4M,480264
+torio/lib/_torio_ffmpeg5.so,sha256=VOZEJc3_WzVRFV-jR_z4p75WNO_PvRgx8CVCQlOVcmU,480264
+torio/lib/_torio_ffmpeg4.so,sha256=IE6Ggu5CPUcfpaO9x-C6PDqCDeGarQHyKFK2ISV5oBE,480264
 torio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 torio/io/_streaming_media_encoder.py,sha256=rSTYmHdi7RPJ6YPgAyGJhbQvn4mcxLem3nlnr_ophTs,19722
 torio/io/_streaming_media_decoder.py,sha256=vSylEWAB_JXOW-0E1n0zDM3Q3Vf1jc1-CNpdUSs13XU,34376

{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: cp39-cp39-linux_x86_64