From acf733fb3d626ae47baccf82f5448cb277fb2392 Mon Sep 17 00:00:00 2001 From: b3ni15 Date: Sun, 30 Nov 2025 21:43:20 +0100 Subject: [PATCH] Add Resolution 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 679ce44..bc05b9c 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, Problem +from pyannote.audio.core.task import Specifications, Problem, Resolution # allowloading checkpoints that store these classes (PyTorch 2.6+ weights_only change) -add_safe_globals([TorchVersion, Specifications, Problem]) +add_safe_globals([TorchVersion, Specifications, Problem, Resolution]) # --- FFmpeg és Whisper a saját rendszered pipeline-jához, HF Token kötelező --- if len(sys.argv) < 2: