Spaces:
Running
Running
Update src/utils/text_qa.py
Browse files- 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="
|
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 |
|