Sagar Sanghani commited on
Commit
b78b0a3
·
1 Parent(s): 66bfc35
Files changed (2) hide show
  1. model.py +2 -2
  2. requirements.txt +2 -1
model.py CHANGED
@@ -107,9 +107,9 @@ class Model:
107
  ]
108
 
109
  llm = HuggingFaceEndpoint(
110
- repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
111
  huggingfacehub_api_token=self.token,
112
- temperature=0.1
113
  )
114
 
115
  chat = ChatHuggingFace(llm=llm).bind_tools(tools)
 
107
  ]
108
 
109
  llm = HuggingFaceEndpoint(
110
+ repo_id="meta-llama/Llama-2-7b-chat-hf",
111
  huggingfacehub_api_token=self.token,
112
+ temperature=0
113
  )
114
 
115
  chat = ChatHuggingFace(llm=llm).bind_tools(tools)
requirements.txt CHANGED
@@ -7,4 +7,5 @@ python-dotenv
7
  langchain
8
  langchain-core
9
  langchain-community
10
- langchain-huggingface
 
 
7
  langchain
8
  langchain-core
9
  langchain-community
10
+ langchain-huggingface
11
+ langchain-tavily