NEXAS commited on
Commit
b85e6e6
·
verified ·
1 Parent(s): a59c6a6

Update src/utils/text_qa.py

Browse files
Files changed (1) hide show
  1. src/utils/text_qa.py +1 -1
src/utils/text_qa.py CHANGED
@@ -40,7 +40,7 @@ def set_custom_prompt():
40
  return prompt
41
 
42
 
43
- chat_model = ChatGroq(temperature=0, model_name="mixtral-8x7b-32768",api_key=groq_api_key) # type: ignore
44
  #chat_model = ChatGroq(temperature=0, model_name="Llama2-70b-4096")
45
  #chat_model = ChatOllama(model="llama2", request_timeout=30.0)
46
 
 
40
  return prompt
41
 
42
 
43
+ chat_model = ChatGroq(temperature=0, model_name="llama-3.3-70b-versatile",api_key=groq_api_key) # type: ignore
44
  #chat_model = ChatGroq(temperature=0, model_name="Llama2-70b-4096")
45
  #chat_model = ChatOllama(model="llama2", request_timeout=30.0)
46