Skip to content

GitLab

"examples/pipe_ex_2.cpp" did not exist on "ae5388111631d8796114bddabb4678965dad6a2b"
Unverified Commit a40fe42a authored by Baber Abbasi's avatar Baber Abbasi Committed by GitHub
Browse files

remove cuda device assertion (#2680)

parent 8fe3435a
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
lm_eval/models/vllm_causallms.py
View file @ a40fe42a
...@@ -75,7 +75,6 @@ class VLLM(TemplateLM): ...@@ -75,7 +75,6 @@ class VLLM(TemplateLM):
"Please install vllm via `pip install lm-eval[vllm]` or `pip install -e .[vllm]`" "Please install vllm via `pip install lm-eval[vllm]` or `pip install -e .[vllm]`"
) )
assert "cuda" in device or device is None, "vLLM only supports CUDA"
assert max_length is None or max_model_len is None, ( assert max_length is None or max_model_len is None, (
"Either max_length or max_model_len may be provided, but not both" "Either max_length or max_model_len may be provided, but not both"
) )
......
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