Skip to content

Commit

Permalink
Merge branch 'main' into batched_fw
Browse files Browse the repository at this point in the history
  • Loading branch information
MahmoudAshraf97 authored Oct 2, 2024
2 parents 7cc960f + a36fde5 commit fbdc88e
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions diarize_parallel.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,7 @@
stderr=subprocess.PIPE,
)
# Transcribe the audio file

whisper_model = faster_whisper.WhisperModel(
args.model_name, device=args.device, compute_type=mtypes[args.device]
)
Expand All @@ -138,6 +139,7 @@
del whisper_model, whisper_pipeline
torch.cuda.empty_cache()


# Forced Alignment
alignment_model, alignment_tokenizer = load_alignment_model(
args.device,
Expand Down

0 comments on commit fbdc88e

Please sign in to comment.