Adds DEBUG log message for prompt_reset_on_temperature (#399)

Produce DEBUG log message if prompt_reset_on_temperature threshold is met.
This commit is contained in:
Purfview
2023-08-04 08:06:17 +01:00
committed by GitHub
parent 857be6f621
commit 1ce16652ee

View File

@@ -568,6 +568,13 @@ class WhisperModel:
not options.condition_on_previous_text
or temperature > options.prompt_reset_on_temperature
):
if options.condition_on_previous_text:
self.logger.debug(
"Reset prompt. prompt_reset_on_temperature threshold is met %f > %f",
temperature,
options.prompt_reset_on_temperature,
)
prompt_reset_since = len(all_tokens)
def encode(self, features: np.ndarray) -> ctranslate2.StorageView: