Disable swa memory for gpt-oss with spec (#17517)
This commit is contained in:
@@ -1285,6 +1285,13 @@ class ServerArgs:
|
|||||||
"Disable hybrid SWA memory for GPT-OSS model with trtllm_mha attention backend."
|
"Disable hybrid SWA memory for GPT-OSS model with trtllm_mha attention backend."
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if self.speculative_algorithm is not None:
|
||||||
|
# TODO: fix spec with SWA memory cache
|
||||||
|
self.disable_hybrid_swa_memory = True
|
||||||
|
logger.warning(
|
||||||
|
"Disable hybrid SWA memory for GPT-OSS model with speculative decoding."
|
||||||
|
)
|
||||||
|
|
||||||
quant_method = get_quantization_config(hf_config)
|
quant_method = get_quantization_config(hf_config)
|
||||||
is_mxfp4_quant_format = quant_method == "mxfp4"
|
is_mxfp4_quant_format = quant_method == "mxfp4"
|
||||||
if is_mxfp4_quant_format:
|
if is_mxfp4_quant_format:
|
||||||
|
|||||||
Reference in New Issue
Block a user