Skip to content

GitLab

Unverified Commit 2c0c7c39 authored by ai-jz's avatar ai-jz Committed by GitHub
Browse files

feat(benchmarks): support HF model names in multi-turn benchmark (#27850)

parent e6751188
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
benchmarks/multi_turn/benchmark_serving_multi_turn.py
View file @ 2c0c7c39
...@@ -1429,8 +1429,6 @@ async def main() -> None: ...@@ -1429,8 +1429,6 @@ async def main() -> None:
random.seed(args.seed) random.seed(args.seed)
np.random.seed(args.seed) np.random.seed(args.seed)
if not os.path.exists(args.model):
raise OSError(f"Path does not exist: {args.model}")
logger.info("Loading tokenizer") logger.info("Loading tokenizer")
tokenizer = AutoTokenizer.from_pretrained(args.model) tokenizer = AutoTokenizer.from_pretrained(args.model)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment