vahe1994 il y a 1 an
Parent
commit
1c66b688c3
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      src/petals/server/from_pretrained.py

+ 1 - 1
src/petals/server/from_pretrained.py

@@ -98,7 +98,7 @@ def load_pretrained_block(
             set_module_tensor_to_device(block, param_name, "cpu", value=param, dtype=param.dtype)
             set_module_tensor_to_device(block, param_name, "cpu", value=param, dtype=param.dtype)
 
 
     logger.info(f"Loaded {model_name} block {block_index}")
     logger.info(f"Loaded {model_name} block {block_index}")
-    print(block.self_attn.q_proj.weight)
+    print(block.self_attn.q_proj.qzeros)
     return block
     return block