torchaudio 2.4.1__cp310-cp310-win_amd64.whl → 2.5.1__cp310-cp310-win_amd64.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+cpu'
-git_version = 'e8cbe17769796ce963fbc71b8990f1474774e6d2'
+__version__ = '2.5.1+cpu'
+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=y1mU0446Vy2hHpCwMqRZt1UI4ZXl-C4tJp
 torchaudio/pipelines/_tts/impl.py,sha256=wwrTyTEEkew22AnzB_ZklapGaAstJSUBawhA7bOcGXM,15759
 torchaudio/pipelines/_tts/__init__.py,sha256=WKc5c06b_M9MvEohJZghJJWAL7vXvfwRIkdy85UCh04,442
 torchaudio/pipelines/rnnt_pipeline.py,sha256=S0DLMPbt-lqNBWOcjG5KP2IfU1X_oTv95CVmSjxYJ2g,14129
-torchaudio/pipelines/_squim_pipeline.py,sha256=ChaNVSQwwT8ge6XEDJCppu2W5Cz7Ng1Y9PkHhPlVwRg,7346
+torchaudio/pipelines/_squim_pipeline.py,sha256=Pumv9jYoJ5mjOeauyuK7RmE8j3iVFP4VuMmvz6UqM3I,6436
 torchaudio/pipelines/_source_separation_pipeline.py,sha256=WAjiWSlk73VS985GpKweDAfk2aKwQWI6jnGNwYHiRi8,4333
 torchaudio/pipelines/__init__.py,sha256=oMwOu-1T_ugJmhdaoI5NrCDrUAGrpDOlJQO8h-bLAW4,2847
 torchaudio/models/wav2vec2/utils/import_huggingface.py,sha256=NMK6YrAIDfOw8j1tV-3XTwx_mwbJHvg8ldTrAWRztIM,6080
@@ -65,8 +65,8 @@ torchaudio/models/conv_tasnet.py,sha256=D7Y10sOzLe03gygfN1J5R73SIHkIGVQOkqKQ6Ni3
 torchaudio/models/conformer.py,sha256=gVrOYeJkPlVaX-4eZpVzNUe_r3k7g1Y6NaaQ8JZP-r4,10361
 torchaudio/models/_hdemucs.py,sha256=ipAj7965PO_WEZqQwW1om9gQj90UhQOeU6HU3Lpvzwo,39250
 torchaudio/models/__init__.py,sha256=Gi3UQvxjwTLW9wfKlF42O3Vup70d0bk2x-rZS89ASwI,2080
-torchaudio/lib/libtorchaudio.pyd,sha256=VJ6_VOb1U5i8a44D57j9NhHwyxfjqnCYAOsuWvG0VgM,1475584
-torchaudio/lib/_torchaudio.pyd,sha256=_8EJeXvs0km0qfhBsKV9pDc4LpFG-vBwva5EF84MVIE,748032
+torchaudio/lib/libtorchaudio.pyd,sha256=3wmvRwm2lpB5N-iQjNFZtx_Ix8YCZZut7vXiseH2TuE,1544704
+torchaudio/lib/_torchaudio.pyd,sha256=EI3sAS9lYRE0CG64yl0VL7WI7lxjurpoTV9xAyU8eMk,755712
 torchaudio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 torchaudio/io/_playback.py,sha256=UpPb-m35XUlYL2lybQGXAJAvfmUPT_Kqx4jpYArIAz0,2393
 torchaudio/io/_effector.py,sha256=5Kh7br-ZuLzmoRSVXk5JNQ8NkwcGUiY_mrt7d_1W1eg,12217
@@ -119,22 +119,22 @@ torchaudio/_backend/ffmpeg.py,sha256=b6dr67sWg47uFJxIyWLXCtrdPMlgjYdWmj7n919Ph1M
 torchaudio/_backend/common.py,sha256=h9R29RTTW2lqOiKYByETXfXWsrTH65uHxGDKw3bWj-s,1835
 torchaudio/_backend/backend.py,sha256=CakqB9z_4ZtLsQTyMZbEbB0kTqpUe_gidHgObv9acyQ,1618
 torchaudio/_backend/__init__.py,sha256=2bMTZ3tG2_5nvnqAAmvEnGBInO5a9DdUPNoyXEnX1M0,1692
-torchaudio/version.py,sha256=bB0N0Ww-GSV4X5nTfj40qF1x0MV8tM0u6thVqPCjUtQ,85
+torchaudio/version.py,sha256=qzT06xOUNfQow7o3hF8gdvjQFq0WRh9pGF5qIebVMSg,85
 torchaudio/kaldi_io.py,sha256=acwysr6fASV9IcOTF0AbVPCo_VQTu1M2AOn1SXm3GPE,5217
 torchaudio/__init__.py,sha256=kUWnkwcKERMj7m-B4E826aGZWGa5sJs1DEpLP6LZoEs,945
-torchaudio-2.4.1.dist-info/LICENSE,sha256=MmOOF5kxv-VR6r9nsOZ6E7SD4Wa1jdcmNjSrf4nzlvU,1363
-torchaudio-2.4.1.dist-info/METADATA,sha256=KW_d3s7syRoHFnJAb3CgaqP-AQaiQvL6v3rXSVJitCc,6468
-torchaudio-2.4.1.dist-info/WHEEL,sha256=fsW6--WFfuzX2scefE6JfcSZ5dXg5h59u8lqlpL5uuo,101
-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=MmOOF5kxv-VR6r9nsOZ6E7SD4Wa1jdcmNjSrf4nzlvU,1363
+torchaudio-2.5.1.dist-info/METADATA,sha256=Eg_SzdHyPQCcqwJgMGZpYECHYLGXy7xXu4bEPV36V04,6468
+torchaudio-2.5.1.dist-info/WHEEL,sha256=0ZjvOlAkRhiFz0IEm5kQrC9Db9zGCLzyOcgLl0kpzxU,101
+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=2-7XS7CEZB0-M9-Ls5Tki4v7aXGJiVg7WouAUZjt3XI,8273
 torio/utils/__init__.py,sha256=uQV58SlyikUr6yF4HITASCvuX-_fnzbeDxFRzFucQE4,60
-torio/lib/libtorio_ffmpeg6.pyd,sha256=2UmARBO3KmLSiE35Q9UeWR-ovSw-S8ZLK_V18uXjbUs,964608
-torio/lib/libtorio_ffmpeg5.pyd,sha256=WE38ri2ypvQC929F9FAcPZ2mwOZFrq61CiPeUFww454,964608
-torio/lib/libtorio_ffmpeg4.pyd,sha256=wVf45_CBTlmkq1pKLhEwYsROdrHDQ8husDGh3hEoKeQ,964608
-torio/lib/_torio_ffmpeg6.pyd,sha256=lFgrtrkVFHAg_FOrvSycSJgamv1VSULMjWJFhySG5_8,1731072
-torio/lib/_torio_ffmpeg5.pyd,sha256=Exh8y37HoXhIJ30xDHwU_NByVAREwqM4qEAC2ShOb0c,1731072
-torio/lib/_torio_ffmpeg4.pyd,sha256=Pkj8wJlfiK4ZTxb6e9gPa3oGLPHeOyB_MgOgKrfAcaw,1731072
+torio/lib/libtorio_ffmpeg6.pyd,sha256=t9Kcjpcj9wPQr0NrglBFIqjlPLSGAoynK32fpE-f6K8,964096
+torio/lib/libtorio_ffmpeg5.pyd,sha256=H234GE8eI8Wpi5wXA5a_U9HUVK9yDnNvHFy2piMPGNU,964096
+torio/lib/libtorio_ffmpeg4.pyd,sha256=FqPt-8LM2X7QXEAUc7M2bzXnK3dtBpkDVFaE_8uhcSc,964096
+torio/lib/_torio_ffmpeg6.pyd,sha256=QeIxxdS2zNLyXHGHxGyyd-eSAsYvk4Ig00GFWDlGmIc,1744384
+torio/lib/_torio_ffmpeg5.pyd,sha256=N7Qpk-bu8HaCumFaqkxuvlgfhqHAHUpGoiec4GEbOtY,1744384
+torio/lib/_torio_ffmpeg4.pyd,sha256=9OVwffScuIgU0f5pKi9_4Q8UOn2mwjbohJmZW5VkScQ,1744384
 torio/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 torio/io/_streaming_media_encoder.py,sha256=C4zIasotf7GlkQqtRK3vMCt2aN6FkG6NK2KUw0ZdHHo,20224
 torio/io/_streaming_media_decoder.py,sha256=dx0K8PD2PZY7yRY1G_As-_8-LyQDLdYfRZPW1kmrJg0,35354

{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: cp310-cp310-win_amd64