Remove duplicate code from dbrx.py (#17550)

2025-05-01 20:51:58 +02:00
parent 811a6c0972
commit 04f2cfc894
1 changed files with 1 additions and 5 deletions
--- 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