Skip to content

GitLab

Unverified Commit 04f2cfc8 authored by sstamenk's avatar sstamenk Committed by GitHub
Browse files

Remove duplicate code from dbrx.py (#17550)

parent 811a6c09
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 5 deletions
vllm/model_executor/models/dbrx.py
View file @ 04f2cfc8
...@@ -451,13 +451,9 @@ class DbrxForCausalLM(nn.Module, SupportsPP): ...@@ -451,13 +451,9 @@ class DbrxForCausalLM(nn.Module, SupportsPP):
break break
else: 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): if is_pp_missing_parameter(name, self):
continue continue
# Remapping the name of FP8 kv-scale.
name = maybe_remap_kv_scale_name(name, params_dict) name = maybe_remap_kv_scale_name(name, params_dict)
if name is None: if name is None:
continue continue
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment