vision-models / requirements.txt
JanadaSroor's picture
Upload requirements.txt with huggingface_hub
ba7eef1 verified
# Core ML libraries
torch>=2.0.0
torchvision>=0.15.0
torchaudio>=2.0.0
# Transformers and tokenizers
transformers>=4.21.0
huggingface_hub>=0.15.0
# ONNX conversion and runtime
onnx>=1.12.0
onnxruntime>=1.15.0
onnxruntime-tools>=1.7.0
optimum[onnxruntime]>=1.7.0
# Image processing
Pillow>=9.0.0
# Data processing
numpy>=1.21.0
# Optional: for model visualization
matplotlib>=3.5.0
# Optional: for advanced quantization
onnxconverter-common>=1.12.0