Reformat code with black

This commit is contained in:
Max Ryabinin 2023-09-03 01:09:56 +03:00
parent 4644131086
commit f100915641

View File

@ -65,7 +65,7 @@ def load_pretrained_block(
# dummy load, check that keys match
report = block.load_state_dict(state_dict, strict=False)
report.missing_keys.pop("self_attn.qkv_proj.weight", None) # will be filled later
report.missing_keys.pop("self_attn.qkv_proj.weight", None) # will be filled later
assert not report.missing_keys, f"Some block weights are missing: {report.missing_keys}"
for param_name, _ in block.named_parameters():