BentoML icon indicating copy to clipboard operation
BentoML copied to clipboard

bug: transformers save_model and load_model tasks conflict

Open aarnphm opened this issue 3 years ago • 0 comments

Describe the bug

Transformers save_model currently check for task_name and task_definition to be not None to pickle custom pipelines.

This behaviour should be consistent with load_model

To reproduce

See bentoml/_internal/frameworks/transformers.py#load_model,save_model

Expected behavior

No response

Environment

na

aarnphm avatar Dec 22 '22 01:12 aarnphm