TRaw commited on
Commit
4336a94
·
1 Parent(s): 01ad80f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ from langchain.memory import ConversationBufferMemory
7
  os.environ["FIREWORKS_API_KEY"] = "ku9UYtzjSAATlcAstO8yrB89MzvDqJL3lGIkNgnVZ7URxPxK"
8
 
9
  llm = ChatFireworks(
10
- model="accounts/fireworks/models/llama-v2-13b-chat",
11
- model_kwargs={"temperature": 0.2, "max_tokens": 64, "top_p": 1.0},
12
  )
13
 
14
  def predict(message, history):
 
7
  os.environ["FIREWORKS_API_KEY"] = "ku9UYtzjSAATlcAstO8yrB89MzvDqJL3lGIkNgnVZ7URxPxK"
8
 
9
  llm = ChatFireworks(
10
+ model="accounts/fireworks/models/mistral-7b-instruct-4k",
11
+ model_kwargs={"temperature": 0.2, "max_tokens": 445, "top_p": 0.9},
12
  )
13
 
14
  def predict(message, history):