fix: speaker embedding bug (#1178)

* fix: improve handling of speaker embeddings in transcribe_task

* chore: bump version to 3.4.1

pyproject.toml

@@ -2,7 +2,7 @@
 urls = { repository = "https://github.com/m-bain/whisperx" }
 authors = [{ name = "Max Bain" }]
 name = "whisperx"
-version = "3.4.0"
+version = "3.4.1"
 description = "Time-Accurate Automatic Speech Recognition using Whisper."
 readme = "README.md"
 requires-python = ">=3.9, <3.13"

uv.lock

@@ -2788,7 +2788,7 @@ wheels = [
 
 [[package]]
 name = "whisperx"
-version = "3.4.0"
+version = "3.4.1"
 source = { editable = "." }
 dependencies = [
     { name = "ctranslate2" },

whisperx/transcribe.py

@@ -213,12 +213,19 @@ def transcribe_task(args: dict, parser: argparse.ArgumentParser):
         results = []
         diarize_model = DiarizationPipeline(model_name=diarize_model_name, use_auth_token=hf_token, device=device)
         for result, input_audio_path in tmp_results:
-            diarize_segments, speaker_embeddings = diarize_model(
+            diarize_result = diarize_model(
                 input_audio_path, 
                 min_speakers=min_speakers, 
                 max_speakers=max_speakers, 
                 return_embeddings=return_speaker_embeddings
             )
+
+            if return_speaker_embeddings:
+                diarize_segments, speaker_embeddings = diarize_result
+            else:
+                diarize_segments = diarize_result
+                speaker_embeddings = None
+
             result = assign_word_speakers(diarize_segments, result, speaker_embeddings)
             results.append((result, input_audio_path))
     # >> Write