Skip to content

GitLab

Commit 7fa49671 authored by baberabb's avatar baberabb
Browse files

fix precommint

parent 05326b04
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/models/vllm_causallms.py
View file @ 7fa49671
...@@ -60,7 +60,7 @@ please install vllm via `pip install lm-eval[vllm]` or `pip install -e .[vllm]`" ...@@ -60,7 +60,7 @@ please install vllm via `pip install lm-eval[vllm]` or `pip install -e .[vllm]`"
tensor_parallel_size=int(tensor_parallel_size), tensor_parallel_size=int(tensor_parallel_size),
swap_space=int(swap_space), swap_space=int(swap_space),
quantization=quantization, quantization=quantization,
seed=seed, seed=int(seed),
) )
self.tokenizer = self.model.get_tokenizer() self.tokenizer = self.model.get_tokenizer()
self.batch_size = batch_size self.batch_size = batch_size
......
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