TaskCLIP / requirements.txt
HanningChen
Download weights from HF model repo and use cached paths
6feb3b2
raw
history blame
518 Bytes
# requirements.txt
# --- PyTorch CUDA 11.8 wheels ---
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.0.0+cu118
torchvision==0.15.1+cu118
torchaudio==2.0.1+cu118
# --- Web backend ---
fastapi==0.111.0
uvicorn==0.30.1
jinja2==3.1.4
python-multipart==0.0.9
pydantic==2.7.4
starlette==0.37.2
anyio==4.4.0
# --- Core ---
numpy==1.26.4
pillow==10.3.0
opencv-python-headless==4.10.0.84
# --- Models / inference ---
ultralytics==8.4.3
open-clip-torch==2.24.0
huggingface_hub>=0.24.0
pytorchvideo==0.1.5