add tflite

Ashish Gupta · Ashish Gupta · commit 06a17d262c9a · 2020-03-19T09:42:15.000-07:00
diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py
@@ -24,7 +24,7 @@
 
 LOGGER = logging.getLogger("sagemaker")
 
-NEO_ALLOWED_FRAMEWORKS = set(["mxnet", "tensorflow", "keras", "pytorch", "onnx", "xgboost"])
+NEO_ALLOWED_FRAMEWORKS = set(["mxnet", "tensorflow", "keras", "pytorch", "onnx", "xgboost", "tflite"])
 
 NEO_IMAGE_ACCOUNT = {
     "us-west-1": "710691900526",
