Merge pull request #545 from MichaelDecent/0.4.5.dev1

updated allowed models of GroqModel.py
swarmauri · Sep 26, 2024 · b42d7be · b42d7be
2 parents 6b996ab + 8921c33
commit b42d7be
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkgs/swarmauri/swarmauri/llms/concrete/GroqModel.py b/pkgs/swarmauri/swarmauri/llms/concrete/GroqModel.py
@@ -17,10 +17,13 @@ class GroqModel(LLMBase):
         "llama3-70b-8192",
         "mixtral-8x7b-32768",
         "gemma-7b-it",
-        "gemma-2-9b-it",
         "llama-3.1-70b-versatile",
         "llama-3.1-8b-instant",
         "llava-v1.5-7b-4096-preview",
+        "llama-3.2-1b-preview",
+        "llama-3.2-3b-preview",
+        "llama-3.2-11b-text-preview",
+        "llama-3.2-90b-text-preview",
     ]
     name: str = "gemma-7b-it"
     type: Literal["GroqModel"] = "GroqModel"