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
|