Update llm_response_generator.py
Browse files
llm_response_generator.py
CHANGED
@@ -34,7 +34,8 @@ class LLLResponseGenerator():
|
|
34 |
questionnaire: str,
|
35 |
user_text: str,
|
36 |
openai_model_name: str = "",
|
37 |
-
hf_repo_id: str = "tiiuae/falcon-7b-instruct",
|
|
|
38 |
temperature: float = 0.1,
|
39 |
max_length: int = 128,
|
40 |
) -> str:
|
|
|
34 |
questionnaire: str,
|
35 |
user_text: str,
|
36 |
openai_model_name: str = "",
|
37 |
+
# hf_repo_id: str = "tiiuae/falcon-7b-instruct",
|
38 |
+
hf_repo_id: str = "mistralai/Mistral-7B-Instruct-v0.2",
|
39 |
temperature: float = 0.1,
|
40 |
max_length: int = 128,
|
41 |
) -> str:
|