demoPOC commited on
Commit
24cb98e
·
1 Parent(s): ba9a3ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -95,7 +95,7 @@ def getLLMModel(LLMID):
95
  # model_kwargs={"temperature": 0.2,"max_length": 2500})
96
  llm = Replicate(
97
  model="meta/llama-2-13b-chat:f4e2de70d66816a838a89eeeb621910adffb0dd0baba3976c96980970978018d",
98
- model_kwargs={"temperature": 0.2,"max_length": 7500})
99
  print("LLAMA2 13B LLM Selected")
100
  elif LLMID == 2:
101
  llm = Replicate(
 
95
  # model_kwargs={"temperature": 0.2,"max_length": 2500})
96
  llm = Replicate(
97
  model="meta/llama-2-13b-chat:f4e2de70d66816a838a89eeeb621910adffb0dd0baba3976c96980970978018d",
98
+ model_kwargs={"temperature": 0.2,"max_new_tokens":2500})
99
  print("LLAMA2 13B LLM Selected")
100
  elif LLMID == 2:
101
  llm = Replicate(