Spaces:

reab5555
/

Multiple-Speakers-Personality-Analyzer

Runtime error

reab5555 commited on Aug 7, 2024

Commit

447f417

verified ·

1 Parent(s): 53f7690

Update transcription_diarization.py

Files changed (1) hide show

transcription_diarization.py CHANGED Viewed

@@ -46,7 +46,7 @@ class LazyTranscriptionPipeline:
                 feature_extractor=self.processor.feature_extractor,
                 max_new_tokens=128,
                 chunk_length_s=30,
-                batch_size=16,
                 return_timestamps=True,
                 torch_dtype=torch.float16,
                 device=torch.device("cuda"),

                 feature_extractor=self.processor.feature_extractor,
                 max_new_tokens=128,
                 chunk_length_s=30,
+                batch_size=8,
                 return_timestamps=True,
                 torch_dtype=torch.float16,
                 device=torch.device("cuda"),