Merge pull request #1037 from anarkiwi/ts

pin torchserve version.
IQTLabs · Dec 13, 2023 · ff30193 · ff30193
2 parents 19a1929 + df4f892
commit ff30193
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_torchserve.sh b/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

diff --git a/torchserve-cuda.yml b/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:

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