Skip to content

Commit 5c43694

Browse files
authored
fix mmstart benchmrk (#334)
1 parent e7b3716 commit 5c43694

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

benchmarks/benchmarker/base.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,8 @@ def run(
132132
max_new_tokens (int): Maximum number of new tokens to generate, default is 2048
133133
num_runs (int): The number of times to run this benchmark, default is 1. You can set it to a larger number if you want to get more stable results.
134134
"""
135+
if not host.startswith(("http://", "https://")):
136+
host = f"http://{host}"
135137
# Initialize backend
136138
sglang_args = Namespace(host=host, port=port, backend="srt-no-parallel")
137139
set_default_backend(select_sglang_backend(sglang_args))

benchmarks/benchmarker/mmstar.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def create_sgl_function(self):
177177
def run(self, *args, **kwargs):
178178
"""Run benchmark and clean up cache directory."""
179179
try:
180-
super().run(*args, **kwargs)
180+
return super().run(*args, **kwargs)
181181
finally:
182182
# Clean up cache directory
183183
if self.cache_dir and os.path.exists(self.cache_dir):

0 commit comments

Comments
 (0)