diff --git a/python/sglang/srt/models/mllama4.py b/python/sglang/srt/models/mllama4.py index e452cb2d4..bca9e7cc3 100644 --- a/python/sglang/srt/models/mllama4.py +++ b/python/sglang/srt/models/mllama4.py @@ -710,7 +710,7 @@ class Llama4ForConditionalGeneration(nn.Module): """Handle scale parameter remapping. Returns True if handled.""" if "scale" in name and "expert" not in name: remapped_name = maybe_remap_kv_scale_name(name, params_dict) - return remapped_name is not None and remapped_name != name + return remapped_name != name return False def _handle_stacked_params(