Reformat code with black

9 months ago · f100915641
parent 4644131086
commit f100915641
1 changed files with 1 additions and 1 deletions
--- a/src/petals/server/from_pretrained.py
+++ b/src/petals/server/from_pretrained.py
@ -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():