diff --git a/deploy-model-with-torchserve/requirements.txt b/deploy-model-with-torchserve/requirements.txt index 77ab34b..cf7680b 100644 --- a/deploy-model-with-torchserve/requirements.txt +++ b/deploy-model-with-torchserve/requirements.txt @@ -2,6 +2,6 @@ # That your model handler might need # torch, torchvision, etc are already included in the base image torch-model-archiver==0.12.0 -torch==2.4 +torch==2.8.0 torchvision==0.19 torchserve==0.12.0