zenaight commited on
Commit
d93a1a1
·
1 Parent(s): 10cd85d

new llama test

Browse files
Files changed (1) hide show
  1. 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="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
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