pin torchserve version.

tests/test_torchserve.sh

@@ -16,7 +16,7 @@ torch-model-archiver --force --model-name yolov8n --version 1.0 --serialized-fil
 rm -rf model_store && mkdir model_store
 mv yolov8n.mar model_store/
 # TODO: --runtime nvidia is required for Orin, --gpus all for x86
-docker run -v $(pwd)/model_store:/model_store -p 8080:8080 --rm --name testts --entrypoint timeout -d iqtlabs/torchserve 180s /torchserve/torchserve-entrypoint.sh --models yolov8n=yolov8n.mar
+docker run -v $(pwd)/model_store:/model_store -p 8080:8080 --rm --name testts --entrypoint timeout -d iqtlabs/torchserve:v0.0.4 180s /torchserve/torchserve-entrypoint.sh --models yolov8n=yolov8n.mar
 PRED=$(wget -q --retry-connrefused --retry-on-host-error --body-file=persons.jpg --method=PUT -O- --header='Content-Type: image/jpg' http://127.0.0.1:8080/predictions/yolov8n | jq)
 echo $PRED
 if [ "$PRED" = "" ] ; then

torchserve-cuda.yml

@@ -4,7 +4,7 @@ networks:
 services:
   torchserve:
     restart: always
-    image: iqtlabs/cuda-torchserve:latest
+    image: iqtlabs/cuda-torchserve:v0.0.4
     networks:
       - gamutrf
     ports:

torchserve.yml

@@ -4,7 +4,7 @@ networks:
 services:
   torchserve:
     restart: always
-    image: iqtlabs/torchserve:latest
+    image: iqtlabs/torchserve:v0.0.4
     networks:
       - gamutrf
     ports: