Skip to content

Commit

Permalink
[Bugfix] update should_ignore_layer (#11354)
Browse files Browse the repository at this point in the history
Signed-off-by: George Ohashi <[email protected]>
  • Loading branch information
horheynm authored Dec 21, 2024
1 parent dd2b563 commit 51ff216
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ def should_ignore_layer(layer_name: Optional[str],
# in the safetensors checkpoint. So, we convert the name
# from the fused version to unfused + check to make sure that
# each shard of the fused layer has the same scheme.
if proj_name in FUSED_LAYER_NAME_MAPPING:
if proj_name in FUSED_LAYER_NAME_MAPPING and layer_name not in ignore:
shard_proj_names = FUSED_LAYER_NAME_MAPPING[proj_name]

# Convert fused_name --> [shard_names]
Expand Down

0 comments on commit 51ff216

Please sign in to comment.