File revert_infer_model.patch of Package cloud-native-demos

Overview Repositories Revisions Requests Users Attributes Meta

File revert_infer_model.patch of Package cloud-native-demos

diff --git a/elastic_inference/apps/infer_service.py b/elastic_inference/apps/infer_service.py
index fb0e6ce..b0c10f5 100755
--- a/elastic_inference/apps/infer_service.py
+++ b/elastic_inference/apps/infer_service.py
@@ -53,12 +53,12 @@ class InferServiceApp(CLCNAppBase):
         LOG.info("Output broker host: %s", self.out_broker_host)
 
         self.infer_type = self.get_env("INFER_TYPE", "face")
-        self.model_name = self.get_env("INFER_MODEL_NAME")
+        self.model_name = self.get_env("MODEL_NAME")
 
         # MODEL_PATH env got higher priority
-        path = self.get_env("INFER_MODEL_PATH")
+        path = self.get_env("MODEL_PATH")
         if path is not None and len(path) != 0:
-            self.model_dir = self.get_env("INFER_MODEL_PATH")
+            self.model_dir = self.get_env("MODEL_PATH")
         else:
             self.model_dir = self.get_env("MODEL_DIR")
 
diff --git a/elastic_inference/kubernetes/elastic-inference.yaml.template b/elastic_inference/kubernetes/elastic-inference.yaml.template
index c2bc1a3..1c50763 100644
--- a/elastic_inference/kubernetes/elastic-inference.yaml.template
+++ b/elastic_inference/kubernetes/elastic-inference.yaml.template
@@ -78,8 +78,10 @@ spec:
         image: your-own-registry/ei-inference-service
         imagePullPolicy: Always
         env:
-        - name: INFER_MODEL_NAME
-          value: "face-detection-retail-0005_INT8"
+        - name: MODEL_NAME
+          value: "face-detection-retail-0005"
+        - name: MODEL_PRECISION
+          value: INT8
         - name: INPUT_QUEUE_HOST
           value: ei-redis-svc
         - name: OUTPUT_BROKER_HOST
@@ -133,8 +135,10 @@ spec:
         image: your-own-registry/ei-inference-service
         imagePullPolicy: Always
         env:
-        - name: INFER_MODEL_NAME
-          value: "face-detection-retail-0005_FP32"
+        - name: MODEL_NAME
+          value: "face-detection-retail-0005"
+        - name: MODEL_PRECISION
+          value: FP32
         - name: INPUT_QUEUE_HOST
           value: ei-redis-svc
         - name: OUTPUT_BROKER_HOST
@@ -188,7 +192,7 @@ spec:
         image: your-own-registry/ei-inference-service
         imagePullPolicy: Always
         env:
-        - name: INFER_MODEL_NAME
+        - name: MODEL_NAME
           value: "SqueezeNetSSD-5Class"
         - name: INPUT_QUEUE_HOST
           value: ei-redis-svc
@@ -244,8 +248,10 @@ spec:
         image: your-own-registry/ei-inference-service
         imagePullPolicy: Always
         env:
-        - name: INFER_MODEL_NAME
-          value: "person-vehicle-bike-detection-crossroad-0078_INT8"
+        - name: MODEL_NAME
+          value: "person-vehicle-bike-detection-crossroad-0078"
+        - name: MODEL_PRECISION
+          value: INT8
         - name: INPUT_QUEUE_HOST
           value: ei-redis-svc
         - name: OUTPUT_BROKER_HOST
@@ -300,8 +306,10 @@ spec:
         image: your-own-registry/ei-inference-service
         imagePullPolicy: Always
         env:
-        - name: INFER_MODEL_NAME
-          value: "person-vehicle-bike-detection-crossroad-0078_FP32"
+        - name: MODEL_NAME
+          value: "person-vehicle-bike-detection-crossroad-0078"
+        - name: MODEL_PRECISION
+          value: FP32
         - name: INPUT_QUEUE_HOST
           value: ei-redis-svc
         - name: OUTPUT_BROKER_HOST

Places

File revert_infer_model.patch of Package cloud-native-demos

Places