Remove duplicate code from dbrx.py (#17550)

04f2cfc8 · sstamenk · GitHub · 811a6c09 · 04f2cfc8
Unverified Commit 04f2cfc8 authored May 01, 2025 by sstamenk Committed by GitHub May 01, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 5 deletions

vllm/model_executor/models/dbrx.py vllm/model_executor/models/dbrx.py +1 -5

No files found.
--- 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