TonicsYI-6B-200k / requirements.txt
Tonic's picture
Update requirements.txt
68181d2
raw
history blame
199 Bytes
--extra-index-url https://download.pytorch.org/whl/cu113
torch==2.0.1
torchaudio==2.0.2
torchvision==0.15.2
scipy
sentencepiece
einops
diffusers
transformers
accelerate
bitsandbytes
optimum
auto-gptq