Skip to content
This repository was archived by the owner on Aug 1, 2024. It is now read-only.
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
.gitignore

weights/wav2lip_gan.pth
weights/yolov8n-face.pt
weights/onnx_dereverb_By_FoxJoy/vocals.onnx
Expand Down Expand Up @@ -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
3 changes: 2 additions & 1 deletion core/scene_preprocessor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
3 changes: 3 additions & 0 deletions core/whisperx/asr.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
2 changes: 1 addition & 1 deletion core/whisperx/diarize.py
Original file line number Diff line number Diff line change
Expand Up @@ -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',
):
Expand Down