fix for transformers

bentoml · Dec 23, 2021 · 78f6605 · 78f6605
1 parent 8fc2ef7
commit 78f6605
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bentoml/_internal/frameworks/transformers.py b/bentoml/_internal/frameworks/transformers.py
@@ -728,14 +728,15 @@ def __init__(
         framework: str,
         lm_head: str,
         device: int,
+        name: str,
         resource_quota: t.Optional[t.Dict[str, t.Any]],
         batch_options: t.Optional[t.Dict[str, t.Any]],
         model_store: "ModelStore" = Provide[BentoMLContainer.model_store],
         **pipeline_kwargs: t.Any,
     ):
         in_store_tag = model_store.get(tag).tag
         self._tag = in_store_tag
-        super().__init__(str(in_store_tag), resource_quota, batch_options)
+        super().__init__(name, resource_quota, batch_options)
 
         try:
             transformers.pipelines.check_task(tasks)