# PyTorch CPU versions
--extra-index-url https://download.pytorch.org/whl/cpu
torch==1.9.0+cpu
torchvision==0.10.0+cpu
torchaudio==0.9.0

# Core dependencies
numpy==1.21.6
Pillow==8.3.2

# Detectron2 CPU version
-f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.9/index.html
detectron2==0.6

# OpenCV
opencv-python==4.5.5.64

# Git dependencies

# Image utilities
imutils==0.5.4

# NATTEN CPU version for torch 1.9
-f https://shi-labs.com/natten/wheels/cpu/torch1.9/index.html
natten==0.14.6

# Build tools
cython==0.29.35
setuptools==59.5.0

# Scientific computing
scipy==1.7.3
shapely==1.8.5
h5py==3.7.0
scikit-image==0.19.3

# Deep learning utilities
timm==0.4.12
einops==0.6.1

# Development tools
icecream==2.1.3
wandb==0.13.5
tqdm==4.64.1

# NLP utilities
ftfy==6.1.1
regex==2022.10.31
inflect==6.0.4

# File utilities
gdown==4.5.4
wget==3.2
huggingface_hub==0.11.1

# Training framework
pytorch_lightning==1.8.6

# Web interface
gradio==3.35.2
scikit-learn
diffdist
gdown
tqdm
huggingface_hub
wget
# Job submission
submitit==1.4.5