Skip to content

Commit f3c141b

Browse files
fix gr limit (modelscope#1016)
1 parent 8dccaa4 commit f3c141b

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

swift/ui/app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def run_ui():
5959

6060
port = os.environ.get('WEBUI_PORT', None)
6161
concurrent = {}
62-
if version.parse(gr.__version__) < version.parse('4.0.0'):
62+
if version.parse(gr.__version__) < version.parse('4.0.0') and os.environ.get('MODELSCOPE_ENVIRONMENT') != 'studio':
6363
concurrent = {'concurrency_count': 5}
6464
app.queue(**concurrent).launch(
6565
server_name=os.environ.get('WEBUI_SERVER', None),

swift/ui/llm_infer/runtime.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,8 @@ def do_build_ui(cls, base_tab: Type['BaseUI']):
101101
gr.Textbox(elem_id='log', lines=6, visible=False)
102102

103103
concurrency_limit = {}
104-
if version.parse(gr.__version__) >= version.parse('4.0.0'):
104+
if version.parse(gr.__version__) >= version.parse('4.0.0') and os.environ.get(
105+
'MODELSCOPE_ENVIRONMENT') != 'studio':
105106
concurrency_limit = {'concurrency_limit': 5}
106107
cls.log_event = base_tab.element('show_log').click(Runtime.update_log, [], [cls.element('log')]).then(
107108
Runtime.wait, [base_tab.element('running_tasks')], [cls.element('log')], **concurrency_limit)

0 commit comments

Comments
 (0)