testing newer docker

udaij12 · udaij12 · commit e2bec9a7c9a7 · 2024-09-18T18:30:29.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: pytorch/torchserve-kfs-nightly:latest-cpu
+      image: udaij12/torchserve-kfs-nightly:latest-cpu
       resources:
         limits:
           cpu: "1"
diff --git a/kubernetes/kserve/tests/configs/mnist_v1_gpu.yaml b/kubernetes/kserve/tests/configs/mnist_v1_gpu.yaml
@@ -6,7 +6,7 @@ spec:
   predictor:
     pytorch:
       storageUri: gs://kfserving-examples/models/torchserve/image_classifier/v1
-      image: pytorch/torchserve-kfs-nightly:latest-gpu
+      image: udaij12/torchserve-kfs-nightly:latest-gpu
       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: pytorch/torchserve-kfs-nightly:latest-cpu
+      image: udaij12/torchserve-kfs-nightly:latest-cpu
       resources:
         limits:
           cpu: "1"
diff --git a/kubernetes/kserve/tests/configs/mnist_v2_gpu.yaml b/kubernetes/kserve/tests/configs/mnist_v2_gpu.yaml
@@ -7,7 +7,7 @@ spec:
     pytorch:
       protocolVersion: v2
       storageUri: gs://kfserving-examples/models/torchserve/image_classifier/v2
-      image: pytorch/torchserve-kfs-nightly:latest-gpu
+      image: udaij12/torchserve-kfs-nightly:latest-gpu
       resources:
         limits:
           cpu: "1"
