ubowang commited on
Commit
21de27a
·
verified ·
1 Parent(s): 872402e

Update model card: official GitHub/dataset links, unified training configs, results, citation

Browse files
Files changed (1) hide show
  1. README.md +23 -8
README.md CHANGED
@@ -4,6 +4,8 @@ library_name: transformers
4
  pipeline_tag: text-generation
5
  base_model:
6
  - Qwen/Qwen3-8B
 
 
7
  tags:
8
  - code
9
  - software-engineering
@@ -12,15 +14,17 @@ tags:
12
 
13
  # FIM-Mid-8B
14
 
15
- **FIM-Mid-8B** is the mid-trained checkpoint of the FIM 8B pipeline: `Qwen3-8B` after 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).
16
 
17
- ## Model
18
 
19
- Local path: `models/FIM-Mid-8B/` (checkpoints are gitignored; do not commit them).
20
 
21
- - Base model: `Qwen/Qwen3-8B`
22
- - FIM mid-training: `train/FIM_Midtrain_8B.yaml` (AdamW, lr `1.0e-5`, cosine schedule, warmup ratio `0.1`, weight decay `0.05`, one epoch, sequence length `32768`, bf16)
23
- - Post-training: none — see [TIGER-Lab/FIM-8B](https://huggingface.co/TIGER-Lab/FIM-8B) for the post-trained agent model
 
 
24
 
25
  ## Serve with vLLM
26
 
@@ -29,7 +33,7 @@ Ships the native Qwen3 40960 context (the yarn extension to 163840 was applied a
29
  ```bash
30
  CUDA_VISIBLE_DEVICES=0 \
31
  python -m vllm.entrypoints.openai.api_server \
32
- --model models/FIM-Mid-8B \
33
  --served-model-name FIM-Mid-8B \
34
  --host 127.0.0.1 \
35
  --port 8400 \
@@ -41,4 +45,15 @@ python -m vllm.entrypoints.openai.api_server \
41
 
42
  ## Post-training
43
 
44
- To reproduce FIM-8B, run SWE-Lego trajectory SFT from this checkpoint (LLaMA-Factory, full fine-tuning, lr `1.0e-4`, 4 epochs, cutoff 131072 with yarn rope scaling, `qwen3_nothink` template, error masking enabled).
 
 
 
 
 
 
 
 
 
 
 
 
4
  pipeline_tag: text-generation
5
  base_model:
6
  - Qwen/Qwen3-8B
7
+ datasets:
8
+ - TIGER-Lab/FIM-Midtraining-400K
9
  tags:
10
  - code
11
  - software-engineering
 
14
 
15
  # FIM-Mid-8B
16
 
17
+ [📄 Paper (PDF)](https://github.com/TIGER-AI-Lab/FIM-Midtraining/blob/main/paper.pdf) · [💻 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)
18
 
19
+ **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).
20
 
21
+ 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.
22
 
23
+ ## Training
24
+
25
+ - **Base model**: [`Qwen/Qwen3-8B`](https://huggingface.co/Qwen/Qwen3-8B)
26
+ - **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))
27
+ - **Post-training**: none — see [TIGER-Lab/FIM-8B](https://huggingface.co/TIGER-Lab/FIM-8B) for the post-trained agent model
28
 
29
  ## Serve with vLLM
30
 
 
33
  ```bash
34
  CUDA_VISIBLE_DEVICES=0 \
35
  python -m vllm.entrypoints.openai.api_server \
36
+ --model TIGER-Lab/FIM-Mid-8B \
37
  --served-model-name FIM-Mid-8B \
38
  --host 127.0.0.1 \
39
  --port 8400 \
 
45
 
46
  ## Post-training
47
 
48
+ 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.
49
+
50
+ ## Citation
51
+
52
+ ```bibtex
53
+ @article{wang2026fim,
54
+ title={Function-Aware Fill-in-the-Middle as Mid-Training for Coding Agent Foundation Models},
55
+ author={Wang, Yubo and Liang, Jiarong and Zhang, Yuxuan and Liu, Xuye and Wei, Cong and Zhang, Yuyu and Nie, Ping and Chen, Wenhu},
56
+ journal={arXiv preprint},
57
+ year={2026}
58
+ }
59
+ ```