# torch + torchvision are already in the Docker base image (pytorch/pytorch:2.1.2-cuda12.1)
# Do NOT add torch here - it will conflict with the pre-installed CUDA version
transformers==4.38.2
diffusers==0.30.3
accelerate>=0.24.0
opencv-python>=4.8.0
Pillow>=10.0.0
numpy<2.0.0
scipy>=1.11.0
scikit-image>=0.21.0
matplotlib>=3.7.0
tqdm>=4.66.0
requests>=2.31.0
huggingface-hub==0.24.7
gradio==4.36.1
onnxruntime>=1.16.0
# segment-anything and groundingdino are installed via git in Dockerfile
pytorch-lightning
omegaconf