transformers==4.34.1
vllm
datasets
accelerate
wandb
scipy