fix: support repo/model:version

bentoml · Aug 12, 2024 · 9975858 · 9975858
1 parent cc7bd0a
commit 9975858
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/openllm/model.py b/src/openllm/model.py
@@ -96,6 +96,9 @@ def list_bento(
 ) -> typing.List[BentoInfo]:
     ensure_repo_updated()
 
+    if repo_name is None and tag and "/" in tag:
+        repo_name, tag = tag.split("/", 1)
+
     if repo_name is not None:
         config = load_config()
         if repo_name not in config.repos: