Slackbot
11/30/2022, 12:37 AMJiang
11/30/2022, 8:26 AMpre_model = bentoml.mlflow.import_model(
model_name,
model_uri,
signatures={'predict': {'batchable': True}} <-----
)Jiang
11/30/2022, 8:26 AMpre_model = bentoml.mlflow.import_model(
model_name,
model_uri,
signatures={'nhwc': {'batchable': False}}
)Jiang
11/30/2022, 8:26 AMsanghyub lee
12/01/2022, 12:01 AMMLflow pyfunc model support only the predict method, signatures={'nhwc': {'batchable': False}} is not supported
I think torch version of the model and weights is the problem.
• So I model training and update weight with torch 1.13.0+cu117
• retrieve model with torch 1.13.
• Done well!