Spaces:
Sleeping
Sleeping
Add Granite
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
|
|
| 14 |
# --- Available Models ---
|
| 15 |
# We will use 'instruct' or 'chat' models because they are good following orders
|
| 16 |
AVAILABLE_MODELS = {
|
| 17 |
-
"
|
| 18 |
"Gemma 2 9B Instruct": "google/gemma-2-9b-it",
|
| 19 |
"Llama 3 8B Instruct": "meta-llama/Meta-Llama-3-8B-Instruct",
|
| 20 |
"Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
|
|
|
|
| 14 |
# --- Available Models ---
|
| 15 |
# We will use 'instruct' or 'chat' models because they are good following orders
|
| 16 |
AVAILABLE_MODELS = {
|
| 17 |
+
"IBM": "ibm-granite/granite-4.0-h-1b",
|
| 18 |
"Gemma 2 9B Instruct": "google/gemma-2-9b-it",
|
| 19 |
"Llama 3 8B Instruct": "meta-llama/Meta-Llama-3-8B-Instruct",
|
| 20 |
"Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
|