requirements.txt 221 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
accelerate
codetiming
datasets
chenych's avatar
chenych committed
4
flash-attn>=2.4.3
chenych's avatar
chenych committed
5
6
7
8
9
10
11
12
13
14
liger-kernel
mathruler
numpy
omegaconf
pandas
peft
pillow
pyarrow>=15.0.0
pylatexenc
qwen-vl-utils
chenych's avatar
chenych committed
15
ray[default]
chenych's avatar
v0.3.0  
chenych committed
16
tensordict
chenych's avatar
chenych committed
17
torchdata
chenych's avatar
chenych committed
18
transformers>=4.51.0
chenych's avatar
v0.3.0  
chenych committed
19
vllm>=0.7.3
chenych's avatar
chenych committed
20
wandb