Add GRPO training code export (requirements.txt)
Browse files- requirements.txt +54 -0
requirements.txt
ADDED
|
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Real-search GRPO training & evaluation (tested in conda env: genrl, Python 3.11)
|
| 2 |
+
# Install PyTorch for your CUDA version first, then:
|
| 3 |
+
# pip install -r requirements.txt
|
| 4 |
+
# Optional (recommended on H100): pip install flash-attn --no-build-isolation
|
| 5 |
+
|
| 6 |
+
# Core training
|
| 7 |
+
ray==2.55.1
|
| 8 |
+
sglang==0.5.10.post1
|
| 9 |
+
transformers==5.3.0
|
| 10 |
+
torch==2.9.1
|
| 11 |
+
torchvision==0.24.1
|
| 12 |
+
torchaudio==2.9.1
|
| 13 |
+
triton==3.5.1
|
| 14 |
+
accelerate==1.13.0
|
| 15 |
+
peft==0.19.1
|
| 16 |
+
deepspeed==0.19.0
|
| 17 |
+
|
| 18 |
+
# Config & hub
|
| 19 |
+
hydra-core==1.3.2
|
| 20 |
+
omegaconf==2.3.0
|
| 21 |
+
huggingface_hub==1.14.0
|
| 22 |
+
safetensors==0.7.0
|
| 23 |
+
tokenizers==0.22.2
|
| 24 |
+
datasets==4.8.5
|
| 25 |
+
|
| 26 |
+
# Eval & tools
|
| 27 |
+
openai==2.6.1
|
| 28 |
+
pydantic==2.12.5
|
| 29 |
+
requests==2.32.3
|
| 30 |
+
httpx==0.28.1
|
| 31 |
+
aiohttp==3.13.5
|
| 32 |
+
fastapi==0.136.1
|
| 33 |
+
uvicorn==0.46.0
|
| 34 |
+
tiktoken==0.12.0
|
| 35 |
+
sentencepiece==0.2.1
|
| 36 |
+
|
| 37 |
+
# Utilities
|
| 38 |
+
numpy==1.26.4
|
| 39 |
+
pandas==3.0.3
|
| 40 |
+
pyarrow==24.0.0
|
| 41 |
+
fsspec==2026.2.0
|
| 42 |
+
einops==0.8.2
|
| 43 |
+
tqdm==4.67.3
|
| 44 |
+
PyYAML==6.0.1
|
| 45 |
+
jsonschema==4.26.0
|
| 46 |
+
codetiming==1.4.0
|
| 47 |
+
wandb==0.27.0
|
| 48 |
+
tensorboard==2.20.0
|
| 49 |
+
psutil==7.2.2
|
| 50 |
+
Jinja2==3.1.6
|
| 51 |
+
rich==15.0.0
|
| 52 |
+
|
| 53 |
+
# Optional acceleration (install separately if needed)
|
| 54 |
+
# flash-attn==4.0.0b13
|