From bece0f6c8c5fcc342582d312bb40de8e6243c432 Mon Sep 17 00:00:00 2001 From: b3ni15 Date: Sun, 30 Nov 2025 21:34:09 +0100 Subject: [PATCH] Add support for loading additional checkpoint classes in PyTorch --- main.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 232e591..4003839 100644 --- a/main.py +++ b/main.py @@ -18,7 +18,10 @@ console = Console() # --- Safe-global engedélyezés TorchVersion számára (checkpoint hack) --- from torch.serialization import add_safe_globals from torch.torch_version import TorchVersion -add_safe_globals([TorchVersion]) +from pyannote.audio.core.task import Specifications + +# allowloading checkpoints that store these classes (PyTorch 2.6+ weights_only change) +add_safe_globals([TorchVersion, Specifications]) # --- FFmpeg és Whisper a saját rendszered pipeline-jához, HF Token kötelező --- if len(sys.argv) < 2: @@ -235,4 +238,4 @@ subprocess.run([ "dubbed_output.mp4" ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) -console.print("\n✅ Magyar AI dub kész: dubbed_output.mp4") \ No newline at end of file +console.print("\n✅ Magyar AI dub kész: dubbed_output.mp4")