fix URL judgement

InternLM · Jan 29, 2025 · aa8e797 · aa8e797
1 parent 637435f
commit aa8e797
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmdeploy/serve/gradio/app.py b/lmdeploy/serve/gradio/app.py
@@ -37,7 +37,7 @@ def run(model_path_or_server: str,
         max_log_len (int): Max number of prompt characters or prompt tokens
             being printed in log. Default: Unlimited
     """
-    if ':' in model_path_or_server:
+    if '://' in model_path_or_server:
         from lmdeploy.serve.gradio.api_server_backend import run_api_server
         run_api_server(model_path_or_server, server_name, server_port, batch_size, share=share)
     else: