rename initial_offset to initial_timestamp

MahmoudAshraf97 · Dec 18, 2023 · 3957238 · 3957238
1 parent 9c0ab3c
commit 3957238
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Whisper_Transcription_+_NeMo_Diarization.ipynb b/Whisper_Transcription_+_NeMo_Diarization.ipynb
@@ -714,7 +714,7 @@
         "    )\n",
         "    word_timestamps = filter_missing_timestamps(\n",
         "        result_aligned[\"word_segments\"],\n",
-        "        initial_offset=whisper_results[0].get(\"start\"),\n",
+        "        initial_timestamp=whisper_results[0].get(\"start\"),\n",
         "        final_timestamp=whisper_results[-1].get(\"end\"),\n",
         "    )\n",
         "\n",

diff --git a/diarize.py b/diarize.py
@@ -124,7 +124,7 @@
     )
     word_timestamps = filter_missing_timestamps(
         result_aligned["word_segments"],
-        initial_offset=whisper_results[0].get("start"),
+        initial_timestamp=whisper_results[0].get("start"),
         final_timestamp=whisper_results[-1].get("end"),
     )
     # clear gpu vram

diff --git a/diarize_parallel.py b/diarize_parallel.py
@@ -126,7 +126,7 @@
     )
     word_timestamps = filter_missing_timestamps(
         result_aligned["word_segments"],
-        initial_offset=whisper_results[0].get("start"),
+        initial_timestamp=whisper_results[0].get("start"),
         final_timestamp=whisper_results[-1].get("end"),
     )
     # clear gpu vram