testing specific docker image

udaij12 · udaij12 · commit a32eda0621a7 · 2024-09-20T21:52:38.000Z
diff --git a/kubernetes/kserve/tests/configs/mnist_v1_cpu.yaml b/kubernetes/kserve/tests/configs/mnist_v1_cpu.yaml
@@ -6,7 +6,7 @@ spec:
   predictor:
     pytorch:
       storageUri: gs://kfserving-examples/models/torchserve/image_classifier/v1
-      image: udaij12/torchserve-kfs-nightly@sha256:d82f45d338f09ad0376703ae00637c8cdb9647783001600e6b3436284205a4df
+      image: udaij12/torchserve-kfs-nightly:latest-cpu
       resources:
         limits:
           cpu: "1"
diff --git a/kubernetes/kserve/tests/configs/mnist_v2_cpu.yaml b/kubernetes/kserve/tests/configs/mnist_v2_cpu.yaml
@@ -7,7 +7,7 @@ spec:
     pytorch:
       protocolVersion: v2
       storageUri: gs://kfserving-examples/models/torchserve/image_classifier/v2
-      image: udaij12/torchserve-kfs-nightly@sha256:d82f45d338f09ad0376703ae00637c8cdb9647783001600e6b3436284205a4df
+      image: udaij12/torchserve-kfs-nightly:latest-cpu
       resources:
         limits:
           cpu: "1"
