Spaces:
Runtime error
Runtime error
Update transcription_diarization.py
Browse files
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=
|
48 |
-
chunk_length_s=
|
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,
|