File size: 314 Bytes
13c5606
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
torch==2.6.0
torchvision==0.21.0
transformers==4.52.4
accelerate
deepspeed
datasets
evaluate
peft
safetensors

flashinfer-python>=0.2.12
xformers
triton>=3.0.0

tqdm
numpy
scipy
pandas
scikit-learn
sentencepiece
huggingface-hub
omegaconf
liger_kernel
termcolor
nest_asyncio
ipykernel
math_verify

wandb
tensorboard