Qwen2.5-0.5B-SFT-UltraChat

Qwen2.5-0.5B-SFT-UltraChat

This model is a fine-tuned version of Qwen/Qwen2.5-0.5B trained on the HuggingFaceH4/ultrachat_200k dataset using SFT with LoRA adapters.

Overview

Qwen2.5-0.5B-SFT-UltraChat is a language model optimized using SFT. Supervised Fine-Tuning (SFT) trains the model to follow instructions by learning from high-quality demonstration data.

Key Features

• High-Quality Fine-Tuning: Trained on 197,471 carefully curated conversation examples
• Efficient Training: Uses LoRA (Low-Rank Adaptation) with 4-bit quantization
• Strong Performance: Achieves 64.27% token accuracy on evaluation set
• Optimized for Inference: Available in multiple formats including GGUF quantizations

Model Details

Training Information

Training Configuration

LoRA Configuration

Training Metrics

Dataset

This model was trained on the HuggingFaceH4/ultrachat_200k dataset.

Usage

Quick Start

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model_id = "ermiaazarkhalili/Qwen2.5-0.5B-SFT-UltraChat"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    torch_dtype=torch.bfloat16,
    device_map="auto"
)

messages = [
    {"role": "system", "content": "You are a helpful assistant."},
    {"role": "user", "content": "What are the key principles of effective communication?"}
]

text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer(text, return_tensors="pt").to(model.device)

outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.7, do_sample=True)
response = tokenizer.decode(outputs[0][inputs['input_ids'].shape[1]:], skip_special_tokens=True)
print(response)

Using Pipeline

from transformers import pipeline

generator = pipeline("text-generation", model="ermiaazarkhalili/Qwen2.5-0.5B-SFT-UltraChat", device_map="auto")
messages = [{"role": "user", "content": "Explain quantum computing in simple terms."}]
output = generator(messages, max_new_tokens=256, return_full_text=False)
print(output[0]["generated_text"])

4-bit Quantized Inference

from transformers import AutoModelForCausalLM, BitsAndBytesConfig
import torch

quantization_config = BitsAndBytesConfig(
    load_in_4bit=True,
    bnb_4bit_compute_dtype=torch.bfloat16,
    bnb_4bit_use_double_quant=True,
    bnb_4bit_quant_type="nf4"
)

model = AutoModelForCausalLM.from_pretrained(
    "ermiaazarkhalili/Qwen2.5-0.5B-SFT-UltraChat",
    quantization_config=quantization_config,
    device_map="auto"
)

GGUF Versions

For CPU or mixed CPU/GPU inference, GGUF quantized versions are available at: ermiaazarkhalili/Qwen2.5-0.5B-SFT-UltraChat-GGUF

Using with Ollama

ollama pull hf.co/ermiaazarkhalili/Qwen2.5-0.5B-SFT-UltraChat-GGUF:Q4_K_M
ollama run hf.co/ermiaazarkhalili/Qwen2.5-0.5B-SFT-UltraChat-GGUF:Q4_K_M "Hello!"

Limitations

• Language: Primarily trained on English data
• Knowledge Cutoff: Limited to base model's training data cutoff
• Hallucinations: May generate plausible-sounding but incorrect information
• Context Length: Fine-tuned with 2,048 token limit
• Safety: Not extensively safety-tuned; use with appropriate guardrails

Intended Use

Recommended Uses

• Research on language model fine-tuning
• Educational purposes
• Personal projects
• Prototyping conversational AI

Out-of-Scope Uses

• Production systems without additional safety measures
• Medical, legal, or financial advice
• Generating harmful or misleading content

Training Framework

• TRL: 0.18.1
• Transformers: 4.52.4
• PyTorch: 2.6.0
• Datasets: 3.6.0
• PEFT: 0.15.2

Citation

@misc{ermiaazarkhalili_qwen25_05b_sft_ultrachat,
    author = {ermiaazarkhalili},
    title = {Qwen2.5-0.5B-SFT-UltraChat: Fine-tuned Qwen2.5-0.5B on ultrachat_200k},
    year = {2025},
    publisher = {Hugging Face},
    howpublished = {\url{https://huggingface.co/ermiaazarkhalili/Qwen2.5-0.5B-SFT-UltraChat}}
}

Acknowledgments

• Base model developers at Qwen
• Hugging Face TRL Team for the training library
• Dataset creators and contributors
• Compute Canada / DRAC for HPC resources

Contact

For questions or issues, please open an issue on the model repository.