Added temperature argument

2024-11-13 11:51:02 +01:00
parent 2fa2eca9bf
commit aa37182614
2 changed files with 14 additions and 2 deletions
--- a/sia/main.py
+++ b/sia/main.py
@@ -48,7 +48,8 @@ class Main:
            case "hf":
                self._llm = HfLlmEngine(
                    model_id=self._config.model,
-                    api_token=self._config.api_token
+                    api_token=self._config.api_token,
+                    temperature=self._config.temperature
                )
            case "test":
                self._llm = TestLLM()