docs: update supported granite 3.0 models

Signed-off-by: Angel Luu <[email protected]>
foundation-model-stack · Nov 7, 2024 · df1b1dc · df1b1dc
1 parent 8ba56af
commit df1b1dc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -148,6 +148,9 @@ Example: Train.jsonl
 
 Model Name & Size  | Model Architecture | Full Finetuning | Low Rank Adaptation (i.e. LoRA) | qLoRA(quantized LoRA) | 
 -------------------- | ---------------- | --------------- | ------------------------------- | --------------------- |
+Granite PowerLM 3B   | GraniteForCausalLM | ✅* | ✅* | ✅* |
+Granite 3.0 2B       | GraniteForCausalLM | ✔️* | ✔️* | ✔️* |
+Granite 3.0 8B       | GraniteForCausalLM | ✅* | ✅* | ✔️ |
 Granite 3.0 3B       | GraniteForCausalLM     | ✅* | ✅* | ✅* |
 Granite 3.0 8B       | GraniteForCausalLM     | ✅* | ✅* | ✔️  |
 GraniteMoE 1B        | GraniteMoeForCausalLM  | ✅ | ✅** | ? |