CoolFace
Modelpublic

ermiaazarkhalili/Granite-4.1-8B-SFT-Fable5

sourceHugging Faceapache-2.0updated 2mo agoView on Hugging Face
2likes254downloads
Model Card

Granite-4.1-8B-SFT-Fable5

A LoRA fine-tune of `ibm-granite/granite-4.1-8b`, supervised fine-tuned on ermiaazarkhalili/Fable-5-Complete-2M-Clean (private).

Base model`ibm-granite/granite-4.1-8b`
ArchitectureGraniteForCausalLM
Parameters8.8B
Training dataermiaazarkhalili/Fable-5-Complete-2M-Clean (private)
MethodLoRA supervised fine-tuning via Unsloth + TRL
Licenseapache-2.0 (inherited from the base model)

Usage

python
from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "ermiaazarkhalili/Granite-4.1-8B-SFT-Fable5"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, dtype='auto', device_map='auto')

messages = [{"role": "user", "content": "Explain gradient checkpointing in two sentences."}]
inputs = tokenizer.apply_chat_template(
    messages, add_generation_prompt=True, return_tensors='pt'
).to(model.device)

outputs = model.generate(inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

Training configuration

SettingValue
LoRA rank (r)16
LoRA alpha16
Learning rate0.0002
Epochs2
Effective batch size8 (2 x 4 grad accum)
Max sequence length4096
Base precision4-bit (QLoRA)
Target modulesq_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj

Observed training loss

Measured from our SLURM logs for this configuration. These are training-loss observations only — no downstream benchmark evaluation has been run on this model, so they should not be read as a quality claim.

SLURM jobStepsFirst lossFinal loss
unlabelled94,2542.28340.8070

Limitations

  • —No benchmark evaluation has been run on this checkpoint. The only reported numbers are training-loss observations.
  • —Inherits the biases, knowledge cutoff and failure modes of the base model.
  • —Fine-tuned on a single instruction-following dataset; behaviour outside that distribution is untested.
  • —LoRA adapters were merged into the base weights, so the merged model cannot be detached from this fine-tune.

Reproducing

Trained by notebooks/fable_distillation_granite41-8b_fable_unsloth.ipynb, executed non-interactively with papermill on a SLURM H100 partition (Unsloth + TRL, LoRA).


Card generated from the training run's own configuration and logs by `scripts/generate_hub_model_card.py`.