adityaverma977 commited on
Commit
5c81985
·
1 Parent(s): 30e238f

Remove rate-limited Groq models (llama-3.1 variants, gemma) and use Mixtral as default

Browse files
Files changed (2) hide show
  1. app/groq_client.py +1 -1
  2. app/hf_spaces.py +1 -3
app/groq_client.py CHANGED
@@ -10,7 +10,7 @@ load_dotenv()
10
  _GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
11
  _client = AsyncGroq(api_key=_GROQ_API_KEY) if _GROQ_API_KEY else None
12
 
13
- DEFAULT_DECISION_MODEL = "llama-3.1-8b-instant"
14
  MAX_AGENT_SPEED = 80
15
 
16
 
 
10
  _GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
11
  _client = AsyncGroq(api_key=_GROQ_API_KEY) if _GROQ_API_KEY else None
12
 
13
+ DEFAULT_DECISION_MODEL = "mixtral-8x7b-32768"
14
  MAX_AGENT_SPEED = 80
15
 
16
 
app/hf_spaces.py CHANGED
@@ -43,10 +43,8 @@ KNOWN_SPACES_MODELS = [
43
 
44
  # Groq models (built-in)
45
  GROQ_MODELS = [
46
- {"id": "llama-3.1-8b-instant", "name": "Llama 3.1 8B", "backend": "groq"},
47
- {"id": "llama-3.1-70b-versatile", "name": "Llama 3.1 70B", "backend": "groq"},
48
  {"id": "mixtral-8x7b-32768", "name": "Mixtral 8x7B", "backend": "groq"},
49
- {"id": "gemma-7b-it", "name": "Gemma 7B", "backend": "groq"},
50
  ]
51
 
52
 
 
43
 
44
  # Groq models (built-in)
45
  GROQ_MODELS = [
 
 
46
  {"id": "mixtral-8x7b-32768", "name": "Mixtral 8x7B", "backend": "groq"},
47
+ {"id": "llama2-70b-4096", "name": "Llama 2 70B", "backend": "groq"},
48
  ]
49
 
50