[Bugfix] Preserve auto-detected quant_config for GLM NextN draft model (#22823)
This commit is contained in:
@@ -129,7 +129,8 @@ class Glm4MoeForCausalLMNextN(Glm4MoeForCausalLM):
|
|||||||
self.config = config
|
self.config = config
|
||||||
self.tp_size = get_tensor_model_parallel_world_size()
|
self.tp_size = get_tensor_model_parallel_world_size()
|
||||||
self.needs_quant_draft = (
|
self.needs_quant_draft = (
|
||||||
get_global_server_args().speculative_draft_model_quantization
|
get_global_server_args().speculative_draft_model_quantization is not None
|
||||||
|
or quant_config is not None
|
||||||
)
|
)
|
||||||
quant_config = quant_config if self.needs_quant_draft else None
|
quant_config = quant_config if self.needs_quant_draft else None
|
||||||
self.model = Glm4MoeModelNextN(
|
self.model = Glm4MoeModelNextN(
|
||||||
|
|||||||
Reference in New Issue
Block a user