devbeni/hun_szinkron
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Problem class to safe globals for checkpoint loading in PyTorch

Browse Source
This commit is contained in:
b3ni15
2025-11-30 21:41:24 +01:00
parent bece0f6c8c
commit 69154bb998
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@@ -18,10 +18,10 @@ console = Console()
# --- Safe-global engedélyezés TorchVersion számára (checkpoint hack) --- # --- Safe-global engedélyezés TorchVersion számára (checkpoint hack) ---
from torch.serialization import add_safe_globals from torch.serialization import add_safe_globals
from torch.torch_version import TorchVersion 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) # 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ő --- # --- FFmpeg és Whisper a saját rendszered pipeline-jához, HF Token kötelező ---
if len(sys.argv) < 2: if len(sys.argv) < 2: