requirements.txt

torch==2.2.0
torchaudio==2.2.0
omegaconf==2.3.0
einops==0.8.1
numpy<2
sentencepiece
git+https://github.com/KingNish24/transformers.git@yue-patch
huggingface-hub==0.25.2
bitsandbytes==0.43.3
accelerate==1.11.0
protobuf==3.20.3
gradio==5.42.0
uvicorn
spaces