Unverified Commit 51ff216d authored by George's avatar George Committed by GitHub
Browse files

[Bugfix] update should_ignore_layer (#11354)


Signed-off-by: default avatarGeorge Ohashi <george@neuralmagic.com>
parent dd2b5633
...@@ -30,7 +30,7 @@ def should_ignore_layer(layer_name: Optional[str], ...@@ -30,7 +30,7 @@ def should_ignore_layer(layer_name: Optional[str],
# in the safetensors checkpoint. So, we convert the name # in the safetensors checkpoint. So, we convert the name
# from the fused version to unfused + check to make sure that # from the fused version to unfused + check to make sure that
# each shard of the fused layer has the same scheme. # 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] shard_proj_names = FUSED_LAYER_NAME_MAPPING[proj_name]
# Convert fused_name --> [shard_names] # Convert fused_name --> [shard_names]
......
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