Fix 1 CPU in ParallelTranscription
Browse files- src/vadParallel.py +2 -1
src/vadParallel.py
CHANGED
@@ -93,7 +93,8 @@ class ParallelTranscription(AbstractTranscription):
|
|
93 |
if (cpu_device_count > 1):
|
94 |
merged = self._get_merged_timestamps_parallel(transcription, audio, config, total_duration, cpu_device_count, cpu_parallel_context)
|
95 |
else:
|
96 |
-
|
|
|
97 |
|
98 |
# Split into a list for each device
|
99 |
# TODO: Split by time instead of by number of chunks
|
|
|
93 |
if (cpu_device_count > 1):
|
94 |
merged = self._get_merged_timestamps_parallel(transcription, audio, config, total_duration, cpu_device_count, cpu_parallel_context)
|
95 |
else:
|
96 |
+
timestamp_segments = transcription.get_transcribe_timestamps(audio, config, 0, total_duration)
|
97 |
+
merged = transcription.get_merged_timestamps(timestamp_segments, config, total_duration)
|
98 |
|
99 |
# Split into a list for each device
|
100 |
# TODO: Split by time instead of by number of chunks
|