Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -92,12 +92,12 @@ def getLLMModel(LLMID):
|
|
| 92 |
if LLMID == 1:
|
| 93 |
llm = Replicate(
|
| 94 |
model="a16z-infra/llama13b-v2-chat:df7690f1994d94e96ad9d568eac121aecf50684a0b0963b25a41cc40061269e5",
|
| 95 |
-
model_kwargs={"temperature": 0.2,"max_length":
|
| 96 |
print("LLAMA2 13B LLM Selected")
|
| 97 |
elif LLMID == 2:
|
| 98 |
llm = Replicate(
|
| 99 |
model="replicate/llama-2-70b-chat:2796ee9483c3fd7aa2e171d38f4ca12251a30609463dcfd4cd76703f22e96cdf",
|
| 100 |
-
model_kwargs={"temperature": 0.2,"
|
| 101 |
print("LLAMA2 7B LLM Selected")
|
| 102 |
elif LLMID == 3:
|
| 103 |
llm = Replicate(model="meta/llama-2-7b-chat:8e6975e5ed6174911a6ff3d60540dfd4844201974602551e10e9e87ab143d81e",
|
|
|
|
| 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:
|
| 98 |
llm = Replicate(
|
| 99 |
model="replicate/llama-2-70b-chat:2796ee9483c3fd7aa2e171d38f4ca12251a30609463dcfd4cd76703f22e96cdf",
|
| 100 |
+
model_kwargs={"temperature": 0.2,"max_length": 2500})
|
| 101 |
print("LLAMA2 7B LLM Selected")
|
| 102 |
elif LLMID == 3:
|
| 103 |
llm = Replicate(model="meta/llama-2-7b-chat:8e6975e5ed6174911a6ff3d60540dfd4844201974602551e10e9e87ab143d81e",
|