vilt-word-patch-alignment / requirements.txt
nielsr's picture
nielsr HF staff
Update requirements.txt
3f11810
pytorch_lightning==1.1.4
transformers
Pillow==8.2.0
tqdm==4.56.0
ipdb==0.13.4
numpy==1.19.5
einops==0.3.0
pyarrow==2.0.0
sacred==0.8.2
pandas==1.1.5
timm
einops
git+https://github.com/rwightman/pytorch-image-models.git
git+https://github.com/dandelin/ViLT.git