fix: set model revision correctly (#5158)

* fix: set model revision correctly Signed-off-by: Frost Ming <me@frostming.com>
bentoml · Dec 27, 2024 · 4e5a653 · 4e5a653
1 parent 11a16aa
commit 4e5a653
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/_bentoml_sdk/service/factory.py b/src/_bentoml_sdk/service/factory.py
@@ -407,8 +407,9 @@ def on_load_bento(self, bento: Bento) -> None:
             model_id = info.metadata.get("model_id")  # use the case in bento info
             if not model_id:
                 model_id = info.tag.name.replace("--", "/")
+            revision = info.metadata.get("revision", info.tag.version)
             model.model_id = model_id
-            model.revision = info.tag.version
+            model.revision = revision
         self.bento = bento