diff --git a/.gitignore b/.gitignore index 462208d..294daaf 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,5 @@ +.gitignore + weights/wav2lip_gan.pth weights/yolov8n-face.pt weights/onnx_dereverb_By_FoxJoy/vocals.onnx @@ -167,3 +169,7 @@ cython_debug/ # and can be added to the global gitignore or merged into this file. For a more nuclear # option (not recommended) you can uncomment the following to ignore the entire idea folder. #.idea/ + + +# MacOs Finder +.DS_Store diff --git a/core/scene_preprocessor.py b/core/scene_preprocessor.py index 6a7477c..142b1bf 100644 --- a/core/scene_preprocessor.py +++ b/core/scene_preprocessor.py @@ -7,7 +7,8 @@ from core.temp_manager import TempFileManager from core.dereverb import MDXNetDereverb from deepface import DeepFace -from deepface.commons import distance as dst +#from deepface.commons import distance as dst +from deepface.modules.verification import find_cosine_distance as dst from pydub import AudioSegment import sqlite3 import uuid diff --git a/core/whisperx/asr.py b/core/whisperx/asr.py index 8d37bcf..08e1983 100644 --- a/core/whisperx/asr.py +++ b/core/whisperx/asr.py @@ -86,6 +86,9 @@ def load_model(whisper_arch, 'prepend_punctuations': "\"'“¿([{-", 'append_punctuations': "\"'.。,,!!??::”)]}、", 'suppress_numerals': False, + "max_new_tokens": None, + "clip_timestamps": None, + "hallucination_silence_threshold": None, } if asr_options is not None: diff --git a/core/whisperx/diarize.py b/core/whisperx/diarize.py index e376255..d62e279 100644 --- a/core/whisperx/diarize.py +++ b/core/whisperx/diarize.py @@ -10,7 +10,7 @@ class DiarizationPipeline: def __init__( self, - model_name='pyannote/speaker-diarization@2.1', + model_name='pyannote/speaker-diarization-3.1', use_auth_token=None, device: Optional[Union[str, torch.device]] = 'cpu', ):