A reasoning-focused fine-tune of IBM Granite 4.0 H 350M.
which itself is an Unsloth-optimized version of IBM Granite 4.0 H 350M.
The dataset contains instruction/response pairs focused on reasoning tasks.
This model is only 350M parameters.
Outputs should be verified before use in high-stakes settings.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "Avtrkrb/granite-claude-h-350m"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(model_id)
7
8prompt = "Explain why the sky appears blue."
9
10inputs = tokenizer(prompt, return_tensors="pt")
11
12outputs = model.generate(
13 **inputs,
14 max_new_tokens=512,
15)
16
17print(tokenizer.decode(outputs[0]))
This repository follows the license of the underlying Granite model.