sparsezoo~=1.8.0
numpy>=1.16.3
onnx<1.15.0,>=1.5.0
pydantic~=2.0
requests>=2.0.0
tqdm>=4.0.0
protobuf>=3.12.2
click!=8.0.0,>=7.1.2

[clip]
open_clip_torch==2.20.0
transformers<4.40

[dev]
beautifulsoup4>=4.9.3
black==22.12.0
flake8>=3.8.3
isort>=5.7.0
pytest-rerunfailures>=13.0
ndjson>=0.3.1
wheel>=0.36.2
pytest>=6.0.0
onnxruntime>=1.7.0
flask>=1.0.0
flask-cors>=3.0.0
Pillow>=8.3.2
openai

[docs]
m2r2~=0.2.7
mistune==0.8.4
myst-parser~=0.14.0
rinohtype>=0.4.2
sphinx>=3.4.0
sphinx-copybutton>=0.3.0
sphinx-markdown-tables>=0.0.15
sphinx-multiversion==0.2.4
sphinx-rtd-theme

[image_classification]
torchvision>=0.3.0
opencv-python<=4.6.0.66

[llm]
transformers<4.40
datasets<2.19
accelerate<0.29
seqeval
evaluate

[onnxruntime]
onnxruntime>=1.7.0

[openpifpaf]
openpifpaf==0.13.11
opencv-python<=4.6.0.66
pycocotools>=2.0.6
scipy==1.10.1

[sentence_transformers]
optimum-deepsparse
torch>=1.7.0

[server]
uvicorn>=0.15.0
fastapi<0.111,>=0.100.0
requests>=2.26.0
python-multipart>=0.0.5
prometheus-client>=0.14.1
psutil>=5.9.4
anyio<4.0.0

[torch]
torch>=1.7.0

[transformers]
transformers<4.40
datasets<2.19
accelerate<0.29
seqeval
evaluate

[yolo]
torchvision>=0.3.0
opencv-python<=4.6.0.66

[yolov5]
torchvision>=0.3.0
opencv-python<=4.6.0.66

[yolov8]
torchvision>=0.3.0
opencv-python<=4.6.0.66
ultralytics==8.0.124
