reab5555 commited on
Commit
9cee20d
·
verified ·
1 Parent(s): 57e78d7

Update transcription_diarization.py

Browse files
Files changed (1) hide show
  1. transcription_diarization.py +2 -2
transcription_diarization.py CHANGED
@@ -44,8 +44,8 @@ class LazyTranscriptionPipeline:
44
  model=self.model,
45
  tokenizer=self.processor.tokenizer,
46
  feature_extractor=self.processor.feature_extractor,
47
- max_new_tokens=64,
48
- chunk_length_s=15,
49
  batch_size=2,
50
  return_timestamps=True,
51
  torch_dtype=torch.bfloat16,
 
44
  model=self.model,
45
  tokenizer=self.processor.tokenizer,
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.bfloat16,