tqdm
mlflow<=3.1.4,>=3.0.0
torch<=2.9.1,>=2.1.0
torchvision
cloudpickle
numpy

[mlserver]
mlserver==1.7.1
mlserver_mlflow
