aadnk commited on
Commit
596fd47
1 Parent(s): cbc9717

Fix 1 CPU in ParallelTranscription

Browse files
Files changed (1) hide show
  1. 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
- merged = transcription.get_merged_timestamps(audio, config, total_duration)
 
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