Hide extra key message

9 months ago · 64b16dd308
parent a14ae7334d
commit 64b16dd308
1 changed files with 2 additions and 1 deletions
--- a/src/petals/server/from_pretrained.py
+++ b/src/petals/server/from_pretrained.py
@ -74,7 +74,8 @@ def load_pretrained_block(
            param = param.to(torch_dtype)
        set_module_tensor_to_device(block, param_name, "cpu", value=param, dtype=param.dtype)

-    logger.info(f"Loaded {model_name} block {block_index}, {report}")
+    logger.info(f"Loaded {model_name} block {block_index}")
+    logger.debug(f"Details: {report}")
    return block