From db4ef590da387c6abd32bea5cc2f2388d7534e05 Mon Sep 17 00:00:00 2001 From: Sverre Nystad Date: Wed, 20 Sep 2023 19:31:43 +0200 Subject: [PATCH] Fix: Correct construction of TextGenerator --- src/text_generation/text_generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/text_generation/text_generator.py b/src/text_generation/text_generator.py index fdb2d47..32ceba6 100644 --- a/src/text_generation/text_generator.py +++ b/src/text_generation/text_generator.py @@ -54,7 +54,7 @@ def get_default_text_generator(temperature: float = 0.7, is_llm: bool = True) -> """ assert 0.0 <= temperature <= 1.0, "Temperature must be between 0.0 and 1.0" if is_llm: - return LLM(temperature) + return LLM(temperature=temperature) else: - return Chatbot(temperature) + return Chatbot(temperature=temperature)