From 35c4bc20d9d454f58506b561b6770d3ae4752bf9 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Fri, 12 Jan 2024 14:02:52 -0800 Subject: [PATCH] [Minor] Fix err msg (#2431) --- vllm/worker/worker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vllm/worker/worker.py b/vllm/worker/worker.py index 4f21edbf3f3b1..c2a2ac148085b 100644 --- a/vllm/worker/worker.py +++ b/vllm/worker/worker.py @@ -239,6 +239,6 @@ def _check_if_gpu_supports_dtype(torch_dtype: torch.dtype): raise ValueError( "Bfloat16 is only supported on GPUs with compute capability " f"of at least 8.0. Your {gpu_name} GPU has compute capability " - f"{compute_capability[0]}.{compute_capability[1]}." - f" You can explicitly specify the data type by using the --dtype option, for example: --dtype=half." - ) + f"{compute_capability[0]}.{compute_capability[1]}. " + "You can use float16 instead by explicitly setting the" + "`dtype` flag in CLI, for example: --dtype=half.")