Nanthasit's picture
Fix model card: replace PEFT template with proper description
4c11b74 verified
|
Raw
History Blame Contribute Delete
2.24 kB
metadata
license: apache-2.0
language:
  - en
library_name: peft
pipeline_tag: text-generation
tags:
  - qwen2
  - lora
  - peft
  - sft
  - trl
  - transformers
  - sakthai
  - tool-calling
  - instruct
  - function-calling
  - text-generation
datasets:
  - Nanthasit/sakthai-combined-v5
base_model: Qwen/Qwen2.5-7B-Instruct

SakThai Context 7B — LoRA Adapter

A LoRA fine-tune of Qwen/Qwen2.5-7B-Instruct for structured tool-calling and instruction following, trained on the SakThai tool-calling curriculum.

Model Details

  • Developed by: Nanthasit
  • Base model: Qwen/Qwen2.5-7B-Instruct (7B parameters)
  • Architecture: Qwen2.5 decoder-only transformer + LoRA adapters
  • Fine-tuning method: LoRA (rank=16, alpha=32) via TRL SFTTrainer
  • Training data: Nanthasit/sakthai-combined-v5
  • License: Apache 2.0
  • Inference: BF16 (use transformers with device_map="auto")

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel

base_model = AutoModelForCausalLM.from_pretrained(
    "Qwen/Qwen2.5-7B-Instruct",
    torch_dtype="bfloat16",
    device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct")

model = PeftModel.from_pretrained(base_model, "Nanthasit/sakthai-context-7b-tools")

Chat Template

The model uses Qwen2.5's standard chat template with system/user/assistant roles and supports function-calling via the tools parameter in the tokenizer.

Merged Version

For production inference, use the merged model instead: 👉 Nanthasit/sakthai-context-7b-merged

Intended Use

  • Tool-calling and function-calling agents
  • Structured instruction following
  • Chat and assistant applications requiring external tool use

Training Details

  • Framework: Hugging Face TRL (SFTTrainer)
  • Compute: HF Jobs (T4 GPU)
  • Quantization: 4-bit NF4 for training
  • Dataset size: ~4,000+ tool-calling examples
  • LoRA config: r=16, lora_alpha=32, target_modules=["q_proj", "k_proj", "v_proj", "o_proj", "gate_proj", "up_proj", "down_proj"]