Commit d0dc5bf9 authored by baberabb's avatar baberabb
Browse files

nits

parent 1a0f57d7
...@@ -81,11 +81,8 @@ please install vllm via `pip install lm-eval[vllm]` or `pip install -e .[vllm]`" ...@@ -81,11 +81,8 @@ please install vllm via `pip install lm-eval[vllm]` or `pip install -e .[vllm]`"
self.model = LLM(**self.model_args) self.model = LLM(**self.model_args)
else: else:
self.model_args["worker_use_ray"] = True self.model_args["worker_use_ray"] = True
if tokenizer:
self.tokenizer = tokenizer
else:
self.tokenizer = get_tokenizer( self.tokenizer = get_tokenizer(
pretrained, tokenizer if tokenizer else pretrained,
tokenizer_mode=tokenizer_mode, tokenizer_mode=tokenizer_mode,
trust_remote_code=trust_remote_code, trust_remote_code=trust_remote_code,
tokenizer_revision=tokenizer_revision, tokenizer_revision=tokenizer_revision,
......
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