small-code-coder-1.5b-tools

A LoRA fine-tune of Qwen2.5-Coder-1.5B-Instruct that teaches the model to emit native <tool_call> function calls, so a ≤2B coder model can drive an agentic coding loop.

Built for smolcode — an SLM-optimized agentic coding assistant — for the Hugging Face Build Small hackathon.

Why

Out of the box, small Qwen-Coder models describe tool calls as plain-text JSON instead of emitting the native <tool_call> format that runtimes (Ollama, llama.cpp) parse — which breaks agentic tool-use loops. This fine-tune closes that gap on a tiny (≤2B, Tiny-Titan-class) model.

Training

  • Base: Qwen/Qwen2.5-Coder-1.5B-Instruct
  • Method: bf16 LoRA (r=16, α=32) on attention + MLP projections, assistant-only loss (loss on tool calls + final answers only).
  • Data: NousResearch/hermes-function-calling-v1 (breadth) + synthetic smolcode tool-use trajectories (sharpness on the actual 5 tools), all rendered through the same apply_chat_template(tools=...) used at inference — so the training target is byte-identical to the served prompt.
  • Schedule: 3 epochs, full 2048 sequence length.
  • Hardware: trained on Modal (x86/CUDA); served on NVIDIA DGX Spark (GB10).

Use

Standard Qwen2.5 chat template with tools=. The model responds with <tool_call>{"name": ..., "arguments": ...}</tool_call> when a tool is warranted.

Status — v2

v2 fixes the v1 train/inference template mismatch (v1 hit 0.92 teacher-forced token accuracy but decoded degenerately because it was trained on a hand-rendered Hermes ChatML format, not Qwen's apply_chat_template output). v2 trains and serves through one shared template and is gated on a free-generation tool-call parse-rate eval (≥90% on held-out smolcode prompts) before release — see eval_toolcall.py in the smolcode repo.

License

Apache-2.0 (inherits from the base model).

Downloads last month
52
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 seanpoyner/smolcode-coder-dotnet-1.5b-tools

Quantized
(132)
this model

Dataset used to train seanpoyner/smolcode-coder-dotnet-1.5b-tools