pop block_cls since it's not an actual kwarg

This commit is contained in:
Wing Lian
2023-09-15 15:54:06 -04:00
parent 360788296a
commit aeec7c4688

View File

@@ -870,7 +870,7 @@ class MixFormerSequentialForCausalLM(MixFormerSequentialPreTrainedModel):
# `block_cls` with `legacy` value is for backward compatibility
# `path` key is for backward compatibility
block = copy.deepcopy(block) or {"block_cls": "parallel"}
# block_cls = block.pop("path", None) or block.pop("block_cls", None)
block.pop("path", None) or block.pop("block_cls", None)
block["block_idx"] = block_idx
modules.append(ParallelBlock(config, **block))