PIPer-8B-RL-only / README.md
PIPer-iclr's picture
Update README.md
61d0b97 verified
|
Raw
History Blame Contribute Delete
3.32 kB
---
library_name: transformers
datasets:
- PIPer-iclr/envbench-zeroshot-rl
base_model:
- PIPer-iclr/Qwen3-8B-am
pipeline_tag: text-generation
license: mit
---
# πŸš€ PIPer: On-Device Environment Setup via Online Reinforcement Learning
<div align="center">
[![Models](https://img.shields.io/badge/πŸ€—%20Hugging%20Face-Models-orange.svg)](https://huggingface.co/PIPer-iclr)
[![Dataset](https://img.shields.io/badge/πŸ€—%20Hugging%20Face-Dataset-green.svg)](https://huggingface.co/datasets/PIPer-iclr/envbench-zeroshot-rl)
[![License](https://img.shields.io/badge/License-MIT-red.svg)](LICENSE)
*Democratizing environment setup with on-device sized models that match the performance of much larger proprietary systems*
</div>
## 🎯 Overview
Environment setupβ€”the process of configuring systems to work with specific software projectsβ€”remains a persistent challenge in software engineering. **PIPer** addresses this by training specialized on-device models that can automatically generate correct Bash scripts for environment configuration.
Our approach combines:
- πŸ“š **Supervised Fine-Tuning (SFT)** with executable scripts from larger models
- 🎯 **Reinforcement Learning with Verifiable Rewards (RLVR)** using lightweight proxy LLM-reward
## πŸ† Key Results
| Model | Size | EnvBench avg@5 | Cost per 1M tokens |
|-------|------|----------------|-------------------|
| **PIPer** | 8B | **19.4** | $0.60 |
| GPT-4o | - | 19.4 | $15.00 |
| Qwen3-32B | 32B | 16.2 | $2.00 |
| Qwen3-8B | 8B | 2.6 | $0.60 |
> πŸŽ‰ **PIPer achieves 9Γ— improvement** over its base model while **matching GPT-4o performance** at **25x lower cost**
![Performance vs Cost Analysis](https://github.com/PIPer-iclr/PIPer/raw/main/misc/combined_pass_n_and_cost.png)
## πŸ“¦ Available Artifacts
### πŸ€– Model Checkpoints
| Model | Description | HuggingFace Link |
|-------|-------------|------------------|
| **πŸ… PIPer (Full)** | Complete SFT+RL trained model | [PIPer-iclr/PIPer-8B](https://huggingface.co/PIPer-iclr/PIPer-8B) |
| 🎯 PIPer (RL-only) | RLVR checkpoint only | [PIPer-iclr/PIPer-8B-RL-only](https://huggingface.co/PIPer-iclr/PIPer-8B-RL-only) |
| πŸ“š PIPer (SFT-only) | Supervised fine-tuning only | [PIPer-iclr/PIPer-8B-SFT-only](https://huggingface.co/PIPer-iclr/PIPer-8B-SFT-only) |
### πŸ“Š Datasets
| Dataset | Description | HuggingFace Link |
|---------|-------------|------------------|
| **EnvBench Zero-shot RL** | Training prompts and evaluation data | [PIPer-iclr/envbench-zeroshot-rl](https://huggingface.co/datasets/PIPer-iclr/envbench-zeroshot-rl) |
## πŸ“Š Evaluation Benchmarks
| Benchmark | Description | Metric | Our Result |
|-----------|-------------|---------|------------|
| **EnvBench-Python** | 329 Python repositories | pass@5 | πŸ† **27/329** |
| **Repo2Run** | 420 Python repositories | pass@5 | πŸ† **103/420** |
| **Terminal-Bench** | 80 terminal tasks | pass@10 | **4/80** |
## πŸ™ Acknowledgments
- Built on top of [EnvBench](https://github.com/princeton-nlp/EnvBench) evaluation framework
- Uses [VeRL](https://github.com/volcengine/verl) for efficient RL training
- Leverages [Qwen3](https://huggingface.co/Qwen) model family as base architecture
## πŸ“„ License
This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.