reab5555 commited on
Commit
1ca98e3
·
verified ·
1 Parent(s): a210f99

Update transcription_diarization.py

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