[BugFix] Drop removed get_cache_scale call in HunyuanImage3 diffusion loader (#4892)

Signed-off-by: wangrzneu <wangrzneu@gmail.com>
This commit is contained in:
Renzheng Wang authored and GitHub committed 2026-07-05 09:50:43 +00:00
1 parent 86bdcaf3d5
commit 02d654760f
1 file changed
+2 -7
@@ -2235,13 +2235,8 @@ class HunyuanImage3Model(nn.Module):
# processed with quantization, LoRA, fine-tuning, etc.
if self.config.tie_word_embeddings and "lm_head.weight" in name:
continue
if self.quant_config is not None and (scale_name := self.quant_config.get_cache_scale(name)):
# Loading kv cache scales for compressed-tensors quantization
param = params_dict[scale_name]
weight_loader = getattr(param, "weight_loader", default_weight_loader)
loaded_weight = loaded_weight[0]
weight_loader(param, loaded_weight)
continue
# KV-cache scales are renamed via maybe_remap_kv_scale_name below;
# quant_config.get_cache_scale was removed in vLLM v0.23.0 (see #4810).
is_found = False
for param_name, weight_name, shard_id in stacked_params_mapping: