ArielShadrac commited on
Commit
8c145de
·
1 Parent(s): a45b838

Switch to Groq llama-3.1-8b-instant via LiteLLM

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. requirements.txt +1 -0
app.py CHANGED
@@ -67,10 +67,10 @@ CRITICAL RULES:
67
  """
68
 
69
  def build_agent():
70
- model = InferenceClientModel(
71
- model_id="llama-3.1-8b-instant",
72
- token=os.getenv("HF"),
73
- max_tokens=2048,
74
  temperature=0.1,
75
  )
76
  agent = CodeAgent(
 
67
  """
68
 
69
  def build_agent():
70
+ model = LiteLLMModel(
71
+ model_id="groq/llama-3.1-8b-instant",
72
+ api_key=os.getenv("HF"),
73
+ max_tokens=1024,
74
  temperature=0.1,
75
  )
76
  agent = CodeAgent(
requirements.txt CHANGED
@@ -3,3 +3,4 @@ gradio
3
  requests
4
  pandas
5
  duckduckgo-search
 
 
3
  requests
4
  pandas
5
  duckduckgo-search
6
+ litellm