Share-Captioner / requirements.txt
Lin-Chen's picture
ZeroGPU (#5)
0dc45ee verified
transformers==4.32.0
accelerate==0.24.0
tiktoken==0.5.1
einops==0.7.0
transformers_stream_generator==0.0.4
scipy==1.11.3
torch==2.1.2
torchvision==0.16.2
pillow==10.0.1
matplotlib==3.8.0
sentencepiece
urllib3==1.26.18
timm==1.0.3
spaces