From f1009156412aed14891ac1bb27e3d4092c4932d6 Mon Sep 17 00:00:00 2001 From: Max Ryabinin Date: Sun, 3 Sep 2023 01:09:56 +0300 Subject: [PATCH] Reformat code with black --- src/petals/server/from_pretrained.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/petals/server/from_pretrained.py b/src/petals/server/from_pretrained.py index 3ca8ef1..5a571b7 100644 --- 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():