Skip to content

Commit

Permalink
update should_ignore_layer
Browse files Browse the repository at this point in the history
  • Loading branch information
horheynm committed Dec 20, 2024
1 parent ca5f54a commit 877d522
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 877d522

Please sign in to comment.