Remove duplicate code from dbrx.py (#17550)

This commit is contained in:
sstamenk 2025-05-01 20:51:58 +02:00 committed by GitHub
parent 811a6c0972
commit 04f2cfc894
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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