add temperature param

This commit is contained in:
Apel-sin
2025-04-02 15:36:14 +03:00
parent fcd3b936cf
commit 144414fcf8
3 changed files with 9 additions and 6 deletions
+2 -1
View File
@@ -38,6 +38,7 @@ class WhisperTranscriber:
self.batch_size = config["batch_size"]
self.max_new_tokens = config["max_new_tokens"]
self.return_timestamps = config["return_timestamps"]
self.temperature = config["temperature"]
# Оптимальный тип для тензоров
self.torch_dtype = torch.bfloat16
@@ -155,7 +156,7 @@ class WhisperTranscriber:
# Транскрибация с корректным форматом данных
result = self.asr_pipeline(
{"raw": audio_array, "sampling_rate": sampling_rate},
generate_kwargs={"language": self.language, "max_new_tokens": self.max_new_tokens},
generate_kwargs={"language": self.language, "max_new_tokens": self.max_new_tokens, "temperature": self.temperature},
return_timestamps=self.return_timestamps
)