"vscode:/vscode.git/clone" did not exist on "da746c608fc7acbb4c9763d08c97fcd9584b809b"
Unverified Commit 9208591f authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix: use fp16 dtype for sm75 (#1136)

parent 5d0d40d0
...@@ -148,6 +148,11 @@ class ModelRunner: ...@@ -148,6 +148,11 @@ class ModelRunner:
f"[gpu={self.gpu_id}] Load weight begin. " f"[gpu={self.gpu_id}] Load weight begin. "
f"avail mem={get_available_gpu_memory(self.gpu_id):.2f} GB" f"avail mem={get_available_gpu_memory(self.gpu_id):.2f} GB"
) )
if torch.cuda.get_device_capability()[0] < 8:
logger.info(
"Compute capability below sm80 use float16 due to lack of bfloat16 support."
)
self.server_args.dtype = "float16"
monkey_patch_vllm_dummy_weight_loader() monkey_patch_vllm_dummy_weight_loader()
device_config = DeviceConfig() device_config = DeviceConfig()
......
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