requirements.txt

# Transformers and NLP related packages
transformers==4.30.2
sentencepiece
protobuf

# Torch and associated libraries
torch==2.1.0
torchvision==0.16.0
triton==2.1.0

# Optimizations and acceleration tools
peft
bitsandbytes
accelerate==0.26.1
deepspeed==0.13.1
pynvml==11.5.0