Add paper link and improve model card for Agent-STAR SFT 7B

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +28 -16
README.md CHANGED
@@ -1,34 +1,34 @@
1
  ---
 
2
  library_name: transformers
3
  license: other
4
- base_model: Qwen/Qwen2.5-7B-Instruct
5
  tags:
6
  - llama-factory
7
  - full
8
  - generated_from_trainer
 
 
9
  model-index:
10
  - name: TravelPlanner_Instruct_7B_1128_1K_4Epoch
11
  results: []
12
  ---
13
 
14
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
- should probably proofread and complete it, then remove this comment. -->
16
-
17
  # TravelPlanner_Instruct_7B_1128_1K_4Epoch
18
 
19
- This model is a fine-tuned version of [Qwen/Qwen2.5-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-7B-Instruct) on the travelplanner_1108_5k_filter_1128_1k dataset.
20
 
21
  ## Model description
22
 
23
- More information needed
24
-
25
- ## Intended uses & limitations
26
 
27
- More information needed
 
 
28
 
29
- ## Training and evaluation data
30
 
31
- More information needed
32
 
33
  ## Training procedure
34
 
@@ -44,18 +44,30 @@ The following hyperparameters were used during training:
44
  - gradient_accumulation_steps: 4
45
  - total_train_batch_size: 32
46
  - total_eval_batch_size: 64
47
- - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
48
  - lr_scheduler_type: cosine
49
  - lr_scheduler_warmup_ratio: 0.1
50
  - num_epochs: 4.0
51
 
52
- ### Training results
53
-
54
-
55
-
56
  ### Framework versions
57
 
58
  - Transformers 4.57.1
59
  - Pytorch 2.9.0+cu128
60
  - Datasets 4.0.0
61
  - Tokenizers 0.22.1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ base_model: Qwen/Qwen2.5-7B-Instruct
3
  library_name: transformers
4
  license: other
5
+ pipeline_tag: text-generation
6
  tags:
7
  - llama-factory
8
  - full
9
  - generated_from_trainer
10
+ - agent
11
+ - tool-use
12
  model-index:
13
  - name: TravelPlanner_Instruct_7B_1128_1K_4Epoch
14
  results: []
15
  ---
16
 
 
 
 
17
  # TravelPlanner_Instruct_7B_1128_1K_4Epoch
18
 
19
+ This model is a fine-tuned version of [Qwen/Qwen2.5-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-7B-Instruct) specialized for long-horizon tool-use tasks. It is part of the **Agent-STAR** project, introduced in the paper [Demystifying Reinforcement Learning for Long-Horizon Tool-Using Agents: A Comprehensive Recipe](https://huggingface.co/papers/2603.21972).
20
 
21
  ## Model description
22
 
23
+ This model corresponds to the **STAR-SFT-7B** checkpoint. It was fine-tuned from the Qwen2.5-Instruct base using **1K successful trajectories** generated by DeepSeek-V3 on synthetic queries within the TravelPlanner environment. This model serves as the Supervised Fine-Tuning (SFT) foundation for the reinforcement learning recipes explored in the paper.
 
 
24
 
25
+ - **Paper:** [Demystifying Reinforcement Learning for Long-Horizon Tool-Using Agents: A Comprehensive Recipe](https://huggingface.co/papers/2603.21972)
26
+ - **Repository:** [https://github.com/WxxShirley/Agent-STAR](https://github.com/WxxShirley/Agent-STAR)
27
+ - **Dataset:** [Agent-STAR-TravelDataset](https://huggingface.co/datasets/xxwu/Agent-STAR-TravelDataset)
28
 
29
+ ## Intended uses & limitations
30
 
31
+ The model is designed to act as an autonomous agent capable of long-horizon planning and tool orchestration. It is specifically optimized for the TravelPlanner benchmark, which requires satisfying multifaceted constraints (commonsense and hard constraints) across multi-turn interactions.
32
 
33
  ## Training procedure
34
 
 
44
  - gradient_accumulation_steps: 4
45
  - total_train_batch_size: 32
46
  - total_eval_batch_size: 64
47
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: cosine
49
  - lr_scheduler_warmup_ratio: 0.1
50
  - num_epochs: 4.0
51
 
 
 
 
 
52
  ### Framework versions
53
 
54
  - Transformers 4.57.1
55
  - Pytorch 2.9.0+cu128
56
  - Datasets 4.0.0
57
  - Tokenizers 0.22.1
58
+
59
+ ## Citation
60
+
61
+ If you find this model helpful, please cite:
62
+
63
+ ```bibtex
64
+ @misc{wu2026agentstar,
65
+ title={Demystifying Reinforcement Learning for Long-Horizon Tool-Using Agents: A Comprehensive Recipe},
66
+ author={Xixi Wu and Qianguo Sun and Ruiyang Zhang and Chao Song and Junlong Wu and Yiyan Qi and Hong Cheng},
67
+ year={2026},
68
+ eprint={2603.21972},
69
+ archivePrefix={arXiv},
70
+ primaryClass={cs.LG},
71
+ url={https://arxiv.org/abs/2603.21972},
72
+ }
73
+ ```