ai_chatbot_phi2model_qlora / requirements.txt
Gosula's picture
Create requirements.txt
a7b74eb
accelerate
transformers
trl
git+https://github.com/huggingface/peft.git
einops
torch
datasets
bitsandbytes
scipy