image-search-colmodernvbert / requirements.txt
Ram07's picture
Add ColModernVBert image search app
3d1eb50
raw
history blame contribute delete
277 Bytes
git+https://github.com/illuin-tech/colpali
huggingface_hub>=0.35.3
torch>=2.2.0
torchvision>=0.17.0
transformers>=4.40.2
pillow>=10.3.0
accelerate>=0.29.0
gradio>=4.44.0
datasets>=2.20.0
tqdm>=4.60.0
matplotlib>=3.5.0
# flash-attn>=2.0.0 # Optional: requires CUDA toolkit