Update transcription_diarization.py

transcription_diarization.py

@@ -45,10 +45,10 @@ class LazyTranscriptionPipeline:
                 tokenizer=self.processor.tokenizer,
                 feature_extractor=self.processor.feature_extractor,
                 max_new_tokens=128,
-                chunk_length_s=15,
+                chunk_length_s=30,
                 batch_size=1,
                 return_timestamps=True,
-                torch_dtype=torch.bfloat16,
+                torch_dtype=torch.float16,
                 device=torch.device("cuda"),
                 generate_kwargs={"language": language}
             )