Add pytorchserver image in kfservice-config (kubeflow#227)

magdalenakuhn17 · Jul 9, 2019 · e53849f · e53849f
1 parent 0d840b9
commit e53849f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 17 deletions.
diff --git a/config/default/configmap/kfservice.yaml b/config/default/configmap/kfservice.yaml
@@ -6,26 +6,29 @@ metadata:
 data:
   frameworks: |-
     {
-        "tensorflow" : {
-            "image" : "tensorflow/serving"
+        "tensorflow": {
+            "image": "tensorflow/serving"
         },
-        "sklearn" : {
-            "image" : "gcr.io/kfserving/sklearnserver"
+        "sklearn": {
+            "image": "gcr.io/kfserving/sklearnserver"
         },
-        "xgboost" : {
-            "image" : "gcr.io/kfserving/xgbserver"
+        "xgboost": {
+            "image": "gcr.io/kfserving/xgbserver"
         },
-        "tensorrt" : {
+        "pytorch": {
+            "image": "gcr.io/kfserving/pytorchserver"
+        }
+        "tensorrt": {
             "image": "nvcr.io/nvidia/tensorrtserver"
         }
     }
   credentials: |-
     {
-       "gcs" : {
+       "gcs": {
            "gcsCredentialFileName": "gcloud-application-credentials.json"
        },
-       "s3" : {
+       "s3": {
            "s3AccessKeyIDName": "awsAccessKeyID",
            "s3SecretAccessKeyName": "awsSecretAccessKey"
        }
-    }
+    }
diff --git a/install/v0.1.0/kfserving.yaml b/install/v0.1.0/kfserving.yaml
@@ -505,16 +505,19 @@ data:
     }
   frameworks: |-
     {
-        "tensorflow" : {
-            "image" : "tensorflow/serving"
+        "tensorflow": {
+            "image": "tensorflow/serving"
         },
-        "sklearn" : {
-            "image" : "gcr.io/kfserving/sklearnserver"
+        "sklearn": {
+            "image": "gcr.io/kfserving/sklearnserver"
         },
-        "xgboost" : {
-            "image" : "gcr.io/kfserving/xgbserver"
+        "pytorch": {
+            "image": "gcr.io/kfserving/pytorchserver"
+        }
+        "xgboost": {
+            "image": "gcr.io/kfserving/xgbserver"
         },
-        "tensorrt" : {
+        "tensorrt": {
             "image": "nvcr.io/nvidia/tensorrtserver"
         }
     }