-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements.txt
More file actions
35 lines (31 loc) · 901 Bytes
/
requirements.txt
File metadata and controls
35 lines (31 loc) · 901 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# Core dependencies
torch==2.6.0
torchvision==0.21.0
transformers>=4.35.0
accelerate>=0.25.0
vllm==0.8.4 # Note: Requires manual LoRA bug patch (see README)
openai>=1.52.0 # Required by vLLM for ChatCompletionContentPartInputAudioParam
# VERL framework for GRPO training
# Requires Python >= 3.9, CUDA >= 12.1
# Install VERL from source (see installation instructions below):
# git clone https://github.com/volcengine/verl.git external/verl
# cd external/verl && pip install --no-deps -e . && cd ../..
# Additional dependencies for VERL
datasets>=2.14.0
peft>=0.6.0
trl>=0.7.0
# deepspeed>=0.12.0 # Optional - requires CUDA >= 12.1 toolkit
ray[default]>=2.43.0
# Utilities
numpy>=1.24.0,<2.0.0 # VERL requires numpy<2.0.0
pandas>=2.0.0
tqdm>=4.65.0
wandb>=0.15.0
# Additional VERL dependencies
codetiming
hydra-core
pybind11
pylatexenc
tensorboard
tensordict>=0.8.0,<=0.10.0,!=0.9.0
torchdata