From 69154bb998e9d36e160c0ea9a5f005f2d51adb2d Mon Sep 17 00:00:00 2001 From: b3ni15 Date: Sun, 30 Nov 2025 21:41:24 +0100 Subject: [PATCH] Add Problem class to safe globals for checkpoint loading in PyTorch --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 4003839..679ce44 100644 --- a/main.py +++ b/main.py @@ -18,10 +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 -from pyannote.audio.core.task import Specifications +from pyannote.audio.core.task import Specifications, Problem # allowloading checkpoints that store these classes (PyTorch 2.6+ weights_only change) -add_safe_globals([TorchVersion, Specifications]) +add_safe_globals([TorchVersion, Specifications, Problem]) # --- FFmpeg és Whisper a saját rendszered pipeline-jához, HF Token kötelező --- if len(sys.argv) < 2: