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: