Text Generation
Transformers
Safetensors
qwen2
agent
tool-use
reinforcement-learning
travel-planner
conversational
text-generation-inference
Instructions to use xxwu/Agent-STAR-RL-3B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use xxwu/Agent-STAR-RL-3B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="xxwu/Agent-STAR-RL-3B") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("xxwu/Agent-STAR-RL-3B") model = AutoModelForCausalLM.from_pretrained("xxwu/Agent-STAR-RL-3B") messages = [ {"role": "user", "content": "Who are you?"}, ] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Inference
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use xxwu/Agent-STAR-RL-3B with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "xxwu/Agent-STAR-RL-3B" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "xxwu/Agent-STAR-RL-3B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/xxwu/Agent-STAR-RL-3B
- SGLang
How to use xxwu/Agent-STAR-RL-3B with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "xxwu/Agent-STAR-RL-3B" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "xxwu/Agent-STAR-RL-3B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "xxwu/Agent-STAR-RL-3B" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "xxwu/Agent-STAR-RL-3B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use xxwu/Agent-STAR-RL-3B with Docker Model Runner:
docker model run hf.co/xxwu/Agent-STAR-RL-3B
Add model card metadata, paper link and GitHub repository
#1
by nielsr HF Staff - opened
README.md
CHANGED
|
@@ -1,3 +1,51 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
library_name: transformers
|
| 4 |
+
pipeline_tag: text-generation
|
| 5 |
+
base_model: Qwen/Qwen2.5-3B-Instruct
|
| 6 |
+
tags:
|
| 7 |
+
- agent
|
| 8 |
+
- tool-use
|
| 9 |
+
- reinforcement-learning
|
| 10 |
+
- travel-planner
|
| 11 |
+
---
|
| 12 |
+
|
| 13 |
+
# Agent-STAR-RL-3B
|
| 14 |
+
|
| 15 |
+
This repository contains the **Agent-STAR-RL-3B** model, a 3B parameter Large Language Model fine-tuned for long-horizon tool orchestration tasks. It was introduced in the paper [Demystifying Reinforcement Learning for Long-Horizon Tool-Using Agents: A Comprehensive Recipe](https://huggingface.co/papers/2603.21972).
|
| 16 |
+
|
| 17 |
+
## Model Description
|
| 18 |
+
|
| 19 |
+
Agent-STAR is a unified post-training pipeline consisting of **[Data Synthesis → SFT → RL]**. This specific checkpoint is the RL-tuned version based on the **Qwen2.5-3B-Instruct** backbone, optimized for the [TravelPlanner](https://github.com/OSU-NLP-Group/TravelPlanner/) benchmark.
|
| 20 |
+
|
| 21 |
+
The model was developed to handle complex, multi-turn agentic environments where it must call various tools to satisfy multifaceted constraints. According to the research findings, smaller models like this 3B variant benefit from staged rewards and enhanced exploration during the RL phase to achieve high performance.
|
| 22 |
+
|
| 23 |
+
## Resources
|
| 24 |
+
|
| 25 |
+
- **Paper:** [Demystifying Reinforcement Learning for Long-Horizon Tool-Using Agents: A Comprehensive Recipe](https://huggingface.co/papers/2603.21972)
|
| 26 |
+
- **GitHub Repository:** [WxxShirley/Agent-STAR](https://github.com/WxxShirley/Agent-STAR)
|
| 27 |
+
- **Dataset:** [Agent-STAR-TravelDataset](https://huggingface.co/datasets/xxwu/Agent-STAR-TravelDataset)
|
| 28 |
+
|
| 29 |
+
## Inference
|
| 30 |
+
|
| 31 |
+
To run inference with this model, please refer to the instructions and ReAct-based inference pipeline provided in the [official GitHub repository](https://github.com/WxxShirley/Agent-STAR).
|
| 32 |
+
|
| 33 |
+
## Citation
|
| 34 |
+
|
| 35 |
+
If you find Agent-STAR helpful to your work, please consider citing:
|
| 36 |
+
|
| 37 |
+
```bibtex
|
| 38 |
+
@misc{wu2026agentstar,
|
| 39 |
+
title={Demystifying Reinforcement Learning for Long-Horizon Tool-Using Agents: A Comprehensive Recipe},
|
| 40 |
+
author={Xixi Wu and Qianguo Sun and Ruiyang Zhang and Chao Song and Junlong Wu and Yiyan Qi and Hong Cheng},
|
| 41 |
+
year={2026},
|
| 42 |
+
eprint={2603.21972},
|
| 43 |
+
archivePrefix={arXiv},
|
| 44 |
+
primaryClass={cs.LG},
|
| 45 |
+
url={https://arxiv.org/abs/2603.21972},
|
| 46 |
+
}
|
| 47 |
+
```
|
| 48 |
+
|
| 49 |
+
## Acknowledgements
|
| 50 |
+
|
| 51 |
+
We appreciate the open-sourced [rLLM](https://github.com/rllm-org/rllm/) framework and the authors of [TravelPlanner](https://github.com/OSU-NLP-Group/TravelPlanner) for providing the benchmark and resources that supported this research.
|