From e1ba1321bc6d840eddea6d886fa85a4343ff8a50 Mon Sep 17 00:00:00 2001
From: bojiang <bojiang_@outlook.com>
Date: Mon, 12 Aug 2024 19:12:46 +0800
Subject: [PATCH] fix: support repo/model:version

---
 src/openllm/model.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/openllm/model.py b/src/openllm/model.py
index af798c82c..f46bd322e 100644
--- 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: