Alexhe101 commited on
Commit
81641d6
·
verified ·
1 Parent(s): 43ace5e

Add GRPO training code export (requirements.txt)

Browse files
Files changed (1) hide show
  1. 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