Tiny fix cannot launch nvfp4 checkpoint with bf16 kv cache (#15986)
This commit is contained in:
@@ -1594,9 +1594,9 @@ class ServerArgs:
|
|||||||
)
|
)
|
||||||
self.page_size = 64
|
self.page_size = 64
|
||||||
|
|
||||||
if self.kv_cache_dtype not in ["fp8_e4m3", "fp4_e2m1", "auto"]:
|
if self.kv_cache_dtype not in ["fp8_e4m3", "fp4_e2m1", "bf16", "auto"]:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
"TensorRT-LLM MLA backend only supports kv-cache-dtype of fp8_e4m3, fp4_e2m1, or auto."
|
"TensorRT-LLM MLA backend only supports kv-cache-dtype of fp8_e4m3, fp4_e2m1, bf16, or auto."
|
||||||
)
|
)
|
||||||
|
|
||||||
if (
|
if (
|
||||||
|
|||||||
Reference in New Issue
Block a user