mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-15 07:05:01 +08:00
Remove duplicate code from dbrx.py (#17550)
This commit is contained in:
parent
811a6c0972
commit
04f2cfc894
@ -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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user