From 04f2cfc89455a05bbeca18c9d5e6c55cd146c560 Mon Sep 17 00:00:00 2001 From: sstamenk Date: Thu, 1 May 2025 20:51:58 +0200 Subject: [PATCH] Remove duplicate code from dbrx.py (#17550) --- vllm/model_executor/models/dbrx.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) 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