Spaces:
Runtime error
Runtime error
zenaight commited on
Commit ·
d93a1a1
1
Parent(s): 10cd85d
new llama test
Browse files- app/main.py +1 -1
app/main.py
CHANGED
|
@@ -180,7 +180,7 @@ def get_llm(model_name: str):
|
|
| 180 |
try:
|
| 181 |
# Try smaller Llama model that's under the 10GB limit
|
| 182 |
return HFInferenceLLM(
|
| 183 |
-
model="
|
| 184 |
provider="hf-inference",
|
| 185 |
api_key=hf_token,
|
| 186 |
max_tokens=4000 # Increased token limit
|
|
|
|
| 180 |
try:
|
| 181 |
# Try smaller Llama model that's under the 10GB limit
|
| 182 |
return HFInferenceLLM(
|
| 183 |
+
model="meta-llama/Llama-3.1-8B",
|
| 184 |
provider="hf-inference",
|
| 185 |
api_key=hf_token,
|
| 186 |
max_tokens=4000 # Increased token limit
|