Phoenix21 commited on
Commit
cb9a86a
·
verified ·
1 Parent(s): ccdc701

Update my_memory_logic.py

Browse files
Files changed (1) hide show
  1. my_memory_logic.py +2 -1
my_memory_logic.py CHANGED
@@ -33,7 +33,8 @@ restatement_prompt = ChatPromptTemplate.from_messages([
33
  # 4) Initialize the ChatGroq LLM
34
  # Ensure you have your GROQ_API_KEY set in the environment
35
  restatement_llm = ChatGroq(
36
- model="llama3-70b-8192", # or whichever model
 
37
  groq_api_key=os.environ["GROQ_API_KEY"]
38
  )
39
 
 
33
  # 4) Initialize the ChatGroq LLM
34
  # Ensure you have your GROQ_API_KEY set in the environment
35
  restatement_llm = ChatGroq(
36
+ model="llama3-70b-8192",
37
+ # model="mixtral-8x7b-32768"# or whichever model
38
  groq_api_key=os.environ["GROQ_API_KEY"]
39
  )
40