torch==2.3.1
transformers==4.43.3
accelerate==0.33.0
deepspeed==0.14.0
bitsandbytes==0.42.0
fsspec==2023.9.2
fsspec[http]==2023.9.2
datasets==2.18.0
numpy==1.26.4
vllm==0.5.3.post1
sympy==1.12
antlr4-python3-runtime==4.11.0
fire
tqdm
tensorboardX
sentencepiece
protobuf
editdistance
wget
appdirs
tempdir
multipledispatch
termcolor
tree_sitter>=0.22.0
tree-sitter-python
wandb