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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -98,10 +98,13 @@ def getLLMModel(LLMID):
98
  model_kwargs={"temperature": 0.2,"max_new_tokens":2500})
99
  print("LLAMA2 13B LLM Selected")
100
  elif LLMID == 2:
 
 
 
101
  llm = Replicate(
102
- model="replicate/llama-2-70b-chat:2796ee9483c3fd7aa2e171d38f4ca12251a30609463dcfd4cd76703f22e96cdf",
103
- model_kwargs={"temperature": 0.2,"max_length": 2500})
104
- print("LLAMA2 7B LLM Selected")
105
  elif LLMID == 3:
106
  llm = Replicate(model="meta/llama-2-7b-chat:8e6975e5ed6174911a6ff3d60540dfd4844201974602551e10e9e87ab143d81e",
107
  model_kwargs={"temperature": 0.2,"max_new_tokens":2500})
 
98
  model_kwargs={"temperature": 0.2,"max_new_tokens":2500})
99
  print("LLAMA2 13B LLM Selected")
100
  elif LLMID == 2:
101
+ # llm = Replicate(
102
+ # model="replicate/llama-2-70b-chat:2796ee9483c3fd7aa2e171d38f4ca12251a30609463dcfd4cd76703f22e96cdf",
103
+ # model_kwargs={"temperature": 0.2,"max_length": 2500})
104
  llm = Replicate(
105
+ model="meta/llama-2-70b-chat:02e509c789964a7ea8736978a43525956ef40397be9033abf9fd2badfe68c9e3",
106
+ model_kwargs={"temperature": 0.2,"max_new_tokens":2500})
107
+ print("LLAMA2 70B LLM Selected")
108
  elif LLMID == 3:
109
  llm = Replicate(model="meta/llama-2-7b-chat:8e6975e5ed6174911a6ff3d60540dfd4844201974602551e10e9e87ab143d81e",
110
  model_kwargs={"temperature": 0.2,"max_new_tokens":2500})