We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a8f74e1 commit ffd4144Copy full SHA for ffd4144
src/autogluon/cloud/scripts/sagemaker_scripts/tabular_serve.py
@@ -32,7 +32,7 @@ def _save_image_and_update_dataframe_column(bytes):
32
def model_fn(model_dir):
33
"""loads model from previously saved artifact"""
34
model = TabularPredictor.load(model_dir)
35
- model.persist_models()
+ model.persist()
36
globals()["column_names"] = model.original_features
37
38
return model
0 commit comments