demoPOC commited on
Commit
c82cef9
·
1 Parent(s): e716324

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -92,7 +92,7 @@ def getLLMModel(LLMID):
92
  if LLMID == 1:
93
  llm = Replicate(
94
  model="a16z-infra/llama13b-v2-chat:df7690f1994d94e96ad9d568eac121aecf50684a0b0963b25a41cc40061269e5",
95
- model_kwargs={"temperature": 0.2,"max_new_tokens":2500})
96
  print("LLAMA2 13B LLM Selected")
97
  elif LLMID == 2:
98
  llm = Replicate(
 
92
  if LLMID == 1:
93
  llm = Replicate(
94
  model="a16z-infra/llama13b-v2-chat:df7690f1994d94e96ad9d568eac121aecf50684a0b0963b25a41cc40061269e5",
95
+ model_kwargs={"temperature": 0.2,"max_length":2500})
96
  print("LLAMA2 13B LLM Selected")
97
  elif LLMID == 2:
98
  llm = Replicate(