jsmanrique commited on
Commit
5f2c033
·
verified ·
1 Parent(s): 857f046

Add Granite

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- "Gemma 3 1B": "litert-community/Gemma3-1B-IT",
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",