demoPOC commited on
Commit
b1d0396
·
1 Parent(s): ad5b8d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -90,8 +90,11 @@ def getLLMModel(LLMID):
90
  # else:
91
  # llm = LlamaCpp(
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:
 
90
  # else:
91
  # llm = LlamaCpp(
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
  llm = Replicate(
97
+ model="meta/llama-2-13b-chat:f4e2de70d66816a838a89eeeb621910adffb0dd0baba3976c96980970978018d",
98
  model_kwargs={"temperature": 0.2,"max_length": 2500})
99
  print("LLAMA2 13B LLM Selected")
100
  elif LLMID == 2: