Persist models (#76)

Co-authored-by: Weisu Yin <[email protected]>
autogluon · May 15, 2023 · 253a750 · 253a750
1 parent 054878f
commit 253a750
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/autogluon/cloud/scripts/sagemaker_scripts/tabular_serve.py b/src/autogluon/cloud/scripts/sagemaker_scripts/tabular_serve.py
@@ -31,6 +31,7 @@ def _save_image_and_update_dataframe_column(bytes):
 def model_fn(model_dir):
     """loads model from previously saved artifact"""
     model = TabularPredictor.load(model_dir)
+    model.persist_models()
     globals()["column_names"] = model.feature_metadata_in.get_features()
 
     return model