Bump torchserve version (kubeflow#2530)

Signed-off-by: Jagadeesh J <[email protected]> Signed-off-by: Jagadeesh J <[email protected]>
magdalenakuhn17 · Nov 17, 2022 · 3d07cbb · 3d07cbb
1 parent 376d008
commit 3d07cbb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/config/runtimes/kustomization.yaml b/config/runtimes/kustomization.yaml
@@ -45,4 +45,4 @@ images:
 
   - name: kserve-torchserve
     newName: pytorch/torchserve-kfs
-    newTag: 0.6.0
+    newTag: 0.6.1
diff --git a/pkg/apis/serving/v1beta1/predictor_torchserve_test.go b/pkg/apis/serving/v1beta1/predictor_torchserve_test.go
@@ -40,7 +40,7 @@ func TestTorchServeValidation(t *testing.T) {
 			spec: PredictorSpec{
 				PyTorch: &TorchServeSpec{
 					PredictorExtensionSpec: PredictorExtensionSpec{
-						RuntimeVersion: proto.String("0.6.0"),
+						RuntimeVersion: proto.String("0.6.1"),
 					},
 				},
 			},
@@ -50,7 +50,7 @@ func TestTorchServeValidation(t *testing.T) {
 			spec: PredictorSpec{
 				PyTorch: &TorchServeSpec{
 					PredictorExtensionSpec: PredictorExtensionSpec{
-						RuntimeVersion: proto.String("0.6.0-gpu"),
+						RuntimeVersion: proto.String("0.6.1-gpu"),
 					},
 				},
 			},
@@ -60,7 +60,7 @@ func TestTorchServeValidation(t *testing.T) {
 			spec: PredictorSpec{
 				PyTorch: &TorchServeSpec{
 					PredictorExtensionSpec: PredictorExtensionSpec{
-						RuntimeVersion: proto.String("0.6.0"),
+						RuntimeVersion: proto.String("0.6.1"),
 						Container: v1.Container{
 							Resources: v1.ResourceRequirements{
 								Limits: v1.ResourceList{constants.NvidiaGPUResourceType: resource.MustParse("1")},
@@ -143,14 +143,14 @@ func TestTorchServeDefaulter(t *testing.T) {
 				PyTorch: &TorchServeSpec{
 					PredictorExtensionSpec: PredictorExtensionSpec{
 						ProtocolVersion: &protocolV1,
-						RuntimeVersion:  proto.String("0.6.0"),
+						RuntimeVersion:  proto.String("0.6.1"),
 					},
 				},
 			},
 			expected: PredictorSpec{
 				PyTorch: &TorchServeSpec{
 					PredictorExtensionSpec: PredictorExtensionSpec{
-						RuntimeVersion:  proto.String("0.6.0"),
+						RuntimeVersion:  proto.String("0.6.1"),
 						ProtocolVersion: &protocolV1,
 						Container: v1.Container{
 							Name: constants.InferenceServiceContainerName,