add temperature param
This commit is contained in:
+2
-1
@@ -3,4 +3,5 @@ __pycache__
|
|||||||
models
|
models
|
||||||
logs
|
logs
|
||||||
client.html
|
client.html
|
||||||
/history/*
|
/history/*
|
||||||
|
config.json.v*
|
||||||
+2
-1
@@ -38,6 +38,7 @@ class WhisperTranscriber:
|
|||||||
self.batch_size = config["batch_size"]
|
self.batch_size = config["batch_size"]
|
||||||
self.max_new_tokens = config["max_new_tokens"]
|
self.max_new_tokens = config["max_new_tokens"]
|
||||||
self.return_timestamps = config["return_timestamps"]
|
self.return_timestamps = config["return_timestamps"]
|
||||||
|
self.temperature = config["temperature"]
|
||||||
|
|
||||||
# Оптимальный тип для тензоров
|
# Оптимальный тип для тензоров
|
||||||
self.torch_dtype = torch.bfloat16
|
self.torch_dtype = torch.bfloat16
|
||||||
@@ -155,7 +156,7 @@ class WhisperTranscriber:
|
|||||||
# Транскрибация с корректным форматом данных
|
# Транскрибация с корректным форматом данных
|
||||||
result = self.asr_pipeline(
|
result = self.asr_pipeline(
|
||||||
{"raw": audio_array, "sampling_rate": sampling_rate},
|
{"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
|
return_timestamps=self.return_timestamps
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
+5
-4
@@ -1,13 +1,14 @@
|
|||||||
{
|
{
|
||||||
"service_port": 5042,
|
"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",
|
"language": "russian",
|
||||||
"enable_history": true,
|
"enable_history": true,
|
||||||
"chunk_length_s": 25,
|
"chunk_length_s": 28,
|
||||||
"batch_size": 20,
|
"batch_size": 8,
|
||||||
"max_new_tokens": 384,
|
"max_new_tokens": 384,
|
||||||
|
"temperature": 0.01,
|
||||||
"return_timestamps": false,
|
"return_timestamps": false,
|
||||||
"audio_rate": 8000,
|
"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"
|
"compand_params": "0.3,1 -90,-90,-70,-50,-40,-15,0,0 -7 0 0.15"
|
||||||
}
|
}
|
||||||
Reference in New Issue
Block a user