FIM-Mid-8B / README.md
ubowang's picture
Add arXiv link (2607.12463)
7f39fb4 verified
|
Raw
History Blame Contribute Delete
3.25 kB
---
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](https://arxiv.org/abs/2607.12463) · [💻 GitHub](https://github.com/TIGER-AI-Lab/FIM-Midtraining) · [🤗 Dataset](https://huggingface.co/datasets/TIGER-Lab/FIM-Midtraining-400K) · [🤗 Collection](https://huggingface.co/collections/TIGER-Lab/fim-midtraining)
**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](https://huggingface.co/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
- **Base model**: [`Qwen/Qwen3-8B`](https://huggingface.co/Qwen/Qwen3-8B)
- **FIM mid-training**: [`midtraining/configs/fim_midtrain.yaml`](https://github.com/TIGER-AI-Lab/FIM-Midtraining/blob/main/midtraining/configs/fim_midtrain.yaml) on [TIGER-Lab/FIM-Midtraining-400K](https://huggingface.co/datasets/TIGER-Lab/FIM-Midtraining-400K) — AdamW, lr `1.0e-5`, cosine schedule, warmup ratio `0.1`, weight decay `0.05`, one epoch, sequence length `32768`, bf16 (as-run copy: [`FIM_Midtrain_8B.yaml`](https://github.com/TIGER-AI-Lab/FIM-Midtraining/blob/main/midtraining/configs/FIM_Midtrain_8B.yaml))
- **Post-training**: none — see [TIGER-Lab/FIM-8B](https://huggingface.co/TIGER-Lab/FIM-8B) for the post-trained agent model
## Serve with vLLM
Ships the native Qwen3 40960 context (the yarn extension to 163840 was applied at post-training time); no overrides needed:
```bash
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`](https://github.com/TIGER-AI-Lab/FIM-Midtraining/blob/main/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/`](https://github.com/TIGER-AI-Lab/FIM-Midtraining/tree/main/posttraining/swe_lego) for the walkthrough.
## Citation
```bibtex
@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}
}
```