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})
|