jaelin215 commited on
Commit
1765ebe
1 Parent(s): d1c4734

Update llm_response_generator.py

Browse files
Files changed (1) hide show
  1. llm_response_generator.py +2 -1
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: