image-classification-finetune / requirements.txt
SupremoUGH's picture
update dockerfile and requirements
fe0ab18 unverified
raw
history blame contribute delete
83 Bytes
torch>=2.0.0
transformers>=4.30.0
Pillow>=9.0.0
datasets>=2.0.0
accelerate>=0.26.0