debug

src/petals/server/from_pretrained.py

@@ -82,6 +82,7 @@ def load_pretrained_block(
     print(model_name,block_prefix,revision,token,cache_dir,max_disk_space)
 
     print("now printing", state_dict)
+    print("block.named_parameters()",block.named_parameters())
     for param_name, _ in block.named_parameters():
         assert param_name in state_dict, f"{param_name} not in state dict"
         param = state_dict[param_name]