Files
vllm/tests/basic_correctness/test_cpu_offload.py
2024-10-22 04:52:14 +00:00

7 lines
175 B
Python

from ..utils import compare_two_settings
def test_cpu_offload():
compare_two_settings("meta-llama/Llama-3.2-1B", [],
["--cpu-offload-gb", "1"])