bupa1018 commited on
Commit
0792287
·
1 Parent(s): 0486180

Create llm.py

Browse files
Files changed (1) hide show
  1. llm.py +3 -0
llm.py ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ def get_groq_llm(model_name, temperature, api_key):
2
+ llm = ChatGroq(model=model_name, temperature=temperature, api_key=api_key)
3
+ return llm