fix: use the model_id in bento (#5087)

Signed-off-by: Frost Ming <me@frostming.com>
bentoml · Nov 15, 2024 · f27377e · f27377e
1 parent 22bbb36
commit f27377e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/_bentoml_sdk/service/factory.py b/src/_bentoml_sdk/service/factory.py
@@ -404,6 +404,10 @@ def on_load_bento(self, bento: Bento) -> None:
             # Replace the model version with the one in the Bento
             if not isinstance(model, HuggingFaceModel):
                 continue
+            model_id = info.metadata.get("model_id")  # use the case in bento info
+            if not model_id:
+                model_id = info.tag.name.replace("--", "/")
+            model.model_id = model_id
             model.revision = info.tag.version
         self.bento = bento