Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -95,7 +95,7 @@ def getLLMModel(LLMID):
|
|
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,"
|
99 |
print("LLAMA2 13B LLM Selected")
|
100 |
elif LLMID == 2:
|
101 |
llm = Replicate(
|
|
|
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_new_tokens":2500})
|
99 |
print("LLAMA2 13B LLM Selected")
|
100 |
elif LLMID == 2:
|
101 |
llm = Replicate(
|