add temperature param
This commit is contained in:
+2
-1
@@ -3,4 +3,5 @@ __pycache__
|
||||
models
|
||||
logs
|
||||
client.html
|
||||
/history/*
|
||||
/history/*
|
||||
config.json.v*
|
||||
+2
-1
@@ -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
|
||||
)
|
||||
|
||||
|
||||
+5
-4
@@ -1,13 +1,14 @@
|
||||
{
|
||||
"service_port": 5042,
|
||||
"model_path": "/mnt/cloud/llm/whisper/whisper-large-v3-russian-ties-podlodka",
|
||||
"model_path": "/home/text-generation/models.whisper/antony-ties-podlodka-v1.2",
|
||||
"language": "russian",
|
||||
"enable_history": true,
|
||||
"chunk_length_s": 25,
|
||||
"batch_size": 20,
|
||||
"chunk_length_s": 28,
|
||||
"batch_size": 8,
|
||||
"max_new_tokens": 384,
|
||||
"temperature": 0.01,
|
||||
"return_timestamps": false,
|
||||
"audio_rate": 8000,
|
||||
"norm_level": "-0.5",
|
||||
"norm_level": "-0.55",
|
||||
"compand_params": "0.3,1 -90,-90,-70,-50,-40,-15,0,0 -7 0 0.15"
|
||||
}
|
||||
Reference in New Issue
Block a user