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