Create model_registry.py
Browse files- model_registry.py +7 -0
model_registry.py
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
MODEL_REGISTRY = {
|
| 2 |
+
"groq_primary": "llama-3.1-8b-instant",
|
| 3 |
+
"groq_backup": "mixtral-8x7b-32768",
|
| 4 |
+
"openai_primary": "gpt-4o-mini",
|
| 5 |
+
"openai_backup": "gpt-3.5-turbo",
|
| 6 |
+
"hf_primary": "meta-llama/Meta-Llama-3-8B-Instruct"
|
| 7 |
+
}
|