reab5555 commited on
Commit
98b6a8f
·
verified ·
1 Parent(s): 7371c07

Update transcription_diarization.py

Browse files
Files changed (1) hide show
  1. transcription_diarization.py +1 -1
transcription_diarization.py CHANGED
@@ -46,7 +46,7 @@ class LazyTranscriptionPipeline:
46
  feature_extractor=self.processor.feature_extractor,
47
  max_new_tokens=128,
48
  chunk_length_s=30,
49
- batch_size=2,
50
  return_timestamps=True,
51
  torch_dtype=torch.float16,
52
  device=torch.device("cuda"),
 
46
  feature_extractor=self.processor.feature_extractor,
47
  max_new_tokens=128,
48
  chunk_length_s=30,
49
+ batch_size=16,
50
  return_timestamps=True,
51
  torch_dtype=torch.float16,
52
  device=torch.device("cuda"),