Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
103 |
-
|
104 |
-
print("LLAMA2
|
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})
|