FIM-Mid-8B / README.md
ubowang's picture
Add arXiv link (2607.12463)
7f39fb4 verified
|
Raw
History Blame Contribute Delete
3.25 kB
metadata
license: apache-2.0
library_name: transformers
pipeline_tag: text-generation
base_model:
  - Qwen/Qwen3-8B
datasets:
  - TIGER-Lab/FIM-Midtraining-400K
tags:
  - code
  - software-engineering
  - fim

FIM-Mid-8B

📄 Paper · 💻 GitHub · 🤗 Dataset · 🤗 Collection

FIM-Mid-8B is the mid-trained checkpoint of the FIM 8B pipeline: Qwen3-8B after function-aware FIM mid-training, before agent post-training. Post-training this checkpoint on SWE-Lego trajectories produces TIGER-Lab/FIM-8B.

It is released for reproducibility and further post-training. The paper deliberately never scores mid-training-only checkpoints — a FIM-only model has degraded instruction-following and cannot be compared fairly against instruction-tuned baselines; every reported gain is one that survives post-training.

Training

Serve with vLLM

Ships the native Qwen3 40960 context (the yarn extension to 163840 was applied at post-training time); no overrides needed:

CUDA_VISIBLE_DEVICES=0 \
python -m vllm.entrypoints.openai.api_server \
  --model TIGER-Lab/FIM-Mid-8B \
  --served-model-name FIM-Mid-8B \
  --host 127.0.0.1 \
  --port 8400 \
  --tensor-parallel-size 1 \
  --max-model-len 40960 \
  --gpu-memory-utilization 0.9 \
  > vllm_fim_mid8b.log 2>&1 &

Post-training

To reproduce FIM-8B, run SWE-Lego trajectory SFT from this checkpoint — the exact config is posttraining/swe_lego/FIM_Posttrain_8B.yaml (LLaMA-Factory, full fine-tuning, lr 1.0e-4, 2 epochs — the official SWE-Lego recipe's 4 overfits this base — cutoff 131072 with yarn rope scaling, qwen3_nothink template, turn_mask enabled), which already points at this repo id. See posttraining/swe_lego/ for the walkthrough.

Citation

@article{wang2026fim,
  title={Function-Aware Fill-in-the-Middle as Mid-Training for Coding Agent Foundation Models},
  author={Wang, Yubo and Liang, Jiarong and Zhang, Yuxuan and Liu, Xuye and Wei, Cong and Zhang, Yuyu and Nie, Ping and Chen, Wenhu},
  journal={arXiv preprint arXiv:2607.12463},
  year={2026}
}