Skip to content

Commit 091a91b

Browse files
committed
update test
Signed-off-by: zhewenli <zhewenli@meta.com>
1 parent bf40fea commit 091a91b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/v1/engine/test_engine_args.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,8 @@ def test_defaults_with_usage_context():
5353
vllm_config: VllmConfig = engine_args.create_engine_config(UsageContext.LLM_CLASS)
5454

5555
from vllm.platforms import current_platform
56+
from vllm.utils.mem_constants import GiB_bytes
5657

57-
GiB_bytes = 1024**3
5858
device_memory = current_platform.get_device_total_memory()
5959
device_name = current_platform.get_device_name().lower()
6060
if device_memory >= 70 * GiB_bytes and "a100" not in device_name:

0 commit comments

Comments
 (0)