diff --git a/vllm/model_executor/models/dbrx.py b/vllm/model_executor/models/dbrx.py index 40c0a73f52d5..9ec245cce189 100644 --- a/vllm/model_executor/models/dbrx.py +++ b/vllm/model_executor/models/dbrx.py @@ -451,13 +451,9 @@ class DbrxForCausalLM(nn.Module, SupportsPP): break else: - # Remapping the name of FP8 kv-scale. - name = maybe_remap_kv_scale_name(name, params_dict) - if name is None: - continue - if is_pp_missing_parameter(name, self): continue + # Remapping the name of FP8 kv-scale. name = maybe_remap_kv_scale_name(name, params_dict) if name is None: continue