Include load models

Nikhil Kulkarni · Nikhil Kulkarni · commit 82c0919c8b18 · 2021-09-24T02:37:57.000-07:00
diff --git a/src/sagemaker_pytorch_serving_container/torchserve.py b/src/sagemaker_pytorch_serving_container/torchserve.py
@@ -175,6 +175,7 @@ def _generate_ts_config_properties():
             }
         }
         user_defined_configuration["models"] = json.dumps(models)
+        user_defined_configuration["load_models"] = f"{DEFAULT_TS_MODEL_NAME}.mar"
 
     custom_configuration = str()
 

Original file line number	Diff line number	Diff line change
`@@ -175,6 +175,7 @@ def _generate_ts_config_properties():`
`175`	`175`	`}`
`176`	`176`	`}`
`177`	`177`	`user_defined_configuration["models"] = json.dumps(models)`
	`178`	`+ user_defined_configuration["load_models"] = f"{DEFAULT_TS_MODEL_NAME}.mar"`
`178`	`179`
`179`	`180`	`custom_configuration = str()`
`180`	`181`