remove cuda device assertion (EleutherAI#2680)

LambdaLabsML · Feb 7, 2025 · a40fe42 · a40fe42
1 parent 8fe3435
commit a40fe42
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/lm_eval/models/vllm_causallms.py b/lm_eval/models/vllm_causallms.py
@@ -75,7 +75,6 @@ def __init__(
                 "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, (
             "Either max_length or max_model_len may be provided, but not both"
         )