VED_caption_generation / Requirements.txt
fangedvampire24's picture
Create Requirements.txt
831b646 verified
torch
torchvision
transformers>=4.30.0
datasets>=2.1.0
accelerate
evaluate
Pillow
numpy