Instructions to use Kwai-Klear/GoLongRL-4B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Kwai-Klear/GoLongRL-4B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="Kwai-Klear/GoLongRL-4B") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("Kwai-Klear/GoLongRL-4B") model = AutoModelForCausalLM.from_pretrained("Kwai-Klear/GoLongRL-4B") 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]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps
- vLLM
How to use Kwai-Klear/GoLongRL-4B with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "Kwai-Klear/GoLongRL-4B" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Kwai-Klear/GoLongRL-4B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/Kwai-Klear/GoLongRL-4B
- SGLang
How to use Kwai-Klear/GoLongRL-4B 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 "Kwai-Klear/GoLongRL-4B" \ --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": "Kwai-Klear/GoLongRL-4B", "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 "Kwai-Klear/GoLongRL-4B" \ --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": "Kwai-Klear/GoLongRL-4B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use Kwai-Klear/GoLongRL-4B with Docker Model Runner:
docker model run hf.co/Kwai-Klear/GoLongRL-4B
✨ GoLongRL-4B
We present GoLongRL, a fully open-source, capability-oriented post-training recipe for long-context reinforcement learning with verifiable rewards (RLVR).
| Resource | Link |
|---|---|
| 📝 Paper | ArXiv (2605.19577) |
| 🤗 HF Paper | GoLongRL |
| 💻 Code | GitHub Repository |
| 📂 Collection | GoLongRL Collection |
| 📊 Dataset | GoLongRL Dataset |
| 📧 Contact | xiao_xuan_zi_666@163.com & suzhenpeng13@163.com |
📌 Overview
Overall performance comparison on long-context benchmarks (DocMath, LongBench-V2, Frames, MRCR, CorpusQA, LBV1-QA).
GoLongRL-4B achieves strong long-context performance at the 4B scale. Our framework combines the following:
Capability-Oriented Dataset (23K samples, 9 task types). Guided by a taxonomy of long-context capabilities, the dataset covers precise retrieval, comprehension, exhaustive retrieval, numerical reasoning, structured extraction, structured matching, graded ranking, sequence ordering, and summarization. Each task is paired with its natural evaluation metric (EM, Accuracy, F1, math_verify, IoU, SubEM, NDCG, Pairwise, ROUGE-L) as the reward function.
TMN-Reweight. To address optimization challenges from heterogeneous rewards, we propose TMN-Reweight, which combines task-level mean normalization for cross-task reward scale alignment with difficulty-adaptive weighting for more reliable advantage estimation. It provides a modest but consistent improvement over vanilla GRPO.
Full Open Release. We publicly release the complete dataset, the four-phase construction pipeline, and all training code.
Key Results
- Under the same vanilla GRPO setup, our dataset alone outperforms the closed-source QwenLong-L1.5 dataset.
- TMN-Reweight further improves average performance to 63.0 at 4B scale, surpassing QwenLong-L1.5 with its specialized AEPO algorithm.
- General capabilities (MMLU-Pro, AIME24/25, GPQA) are preserved or improved, with substantial gains on dialogue memory (LongMemEval +13.6) and agentic memory benchmarks.
🔍 Evaluation
Evaluation uses QwenLong-Benchmarks, covering three capability dimensions:
| Dimension | Benchmarks |
|---|---|
| Long-Context | LongBench-V2, MRCR (≤128K / 128K–512K / 512K–1M), Frames, LongBench QA, DocMath, CorpusQA (≤128K / ≤1M) |
| General | MMLU-Pro, AIME 2024/2025, GPQA-Diamond |
| Memory | BFCL-V4 (memory subset), LongMemEval |
🤝 Citation
@misc{lv2026golongrlcapabilityorientedlongcontext,
title={GoLongRL: Capability-Oriented Long Context Reinforcement Learning with Multitask Alignment},
author={Minxuan Lv and Tiehua Mei and Tanlong Du and Junmin Chen and Zhenpeng Su and Ziyang Chen and Ziqi Wang and Zhennan Wu and Ruotong Pan and jian Liang and Ruiming Tang and Han Li},
year={2026},
eprint={2605.19577},
archivePrefix={arXiv},
primaryClass={cs.CL},
url={https://arxiv.org/abs/2605.19577},
}
- Downloads last month
- 94
Install from pip and serve model
# Install vLLM from pip: pip install vllm# Start the vLLM server: vllm serve "Kwai-Klear/GoLongRL-4B"# Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Kwai-Klear/GoLongRL-4B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'