Views
No views yet
unsloth/granite-4.0-h-micro, optimized using the mlabonne/FineTome-100k dataset.
It leverages IBM’s Granite 4.0 foundation model, enhanced through instruction fine-tuning to improve performance on reasoning, conversational coherence, and instruction-following tasks.unsloth/granite-4.0-h-micromlabonne/FineTome-100k| Parameter | Value |
|---|---|
| Framework | Unsloth |
| Training method | Supervised Fine-Tuning (SFT) |
| Optimizer | AdamW |
| Epochs | 1–3 (depending on convergence) |
| Learning rate | 2e-5 |
| Batch size | 4 (gradient accumulation used) |
| Hardware | A100 / T4 GPU |
| Mixed precision | bf16 |
| Evaluation | Perplexity and instruction accuracy |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "krishanwalia30/granite-4.0-finetome-finetuned" # Replace with your HF repo ID
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
8
9prompt = "Explain why transformers are used in modern NLP."
10inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
11outputs = model.generate(**inputs, max_new_tokens=250, temperature=0.7)
12print(tokenizer.decode(outputs[0], skip_special_tokens=True))