kaana-if-opd

This is an open text-generation model fine-tuned from gyubinc/sft with OPD/TRL.

Training

  • Student/base model: gyubinc/sft
  • Teacher model: gyubinc/autoif
  • Dataset: gyubinc/IF
  • Method: OPD / MiniLLM-style distillation (distill_mode=tml)
  • Precision: bfloat16
  • Training checkpoint: checkpoint-50
  • Max planned steps: 100
  • Saved global step: 50
  • Effective batch size: 128
  • Learning rate: 1e-6
  • Max prompt length: 1024
  • Max completion length: 2048

The source run was interrupted after step 62, and the latest complete saved Hugging Face checkpoint available locally was step 50. This repository contains only inference artifacts and excludes DeepSpeed optimizer state, RNG state, and trainer-resume files.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "riasok/kaana-if-opd"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    torch_dtype="auto",
    device_map="auto",
)

prompt = "Write a concise answer: What is instruction following?"
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

Framework Versions

  • TRL: 0.26.0
  • Transformers: 4.57.1
  • PyTorch: 2.8.0
  • Datasets: 3.6.0
  • Tokenizers: 0.22.2
Downloads last month
1
Safetensors
Model size
2B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Riasok/kaana-if-opd

Base model

gyubinc/sft
Finetuned
(1)
this model