rixty/VoiceCraft
1
0
Fork 0
mirror of https://github.com/jasonppy/VoiceCraft.git synced 2025-06-05 21:49:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

README update, gradio_app.ipynb update, debug print removed

Browse Source
This commit is contained in:
Stepan Zuev
2024-04-05 04:40:57 +03:00
parent bbe3437b8d
commit 94e9f9bd42
3 changed files with 21 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gradio_app.py
View File

@@ -75,9 +75,6 @@ class WhisperxModel:
def load_models(whisper_backend_name, whisper_model_name, alignment_model_name, voicecraft_model_name):
global transcribe_model, align_model, voicecraft_model
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
if alignment_model_name is not None:
align_model = WhisperxAlignModel()
@@ -178,7 +175,6 @@ def align(seed, transcript, audio_path):
} for fragment in fragments["fragments"]]
segments = align_model.align(segments, audio_path)
state = get_transcribe_state(segments)
print(state)
return [
state["transcript_with_start_time"], state["transcript_with_end_time"],