remove_print

Author: xuxinyi389

paddlenlp/transformers/deepseek_v2/modeling_network.py

@@ -393,9 +393,7 @@ def forward(
         # query_states[:, :, :, : self.qk_nope_head_dim] = q_nope
         # query_states[:, :, :, self.qk_nope_head_dim :] = q_pe
 
-        print("k_nope.shape", k_nope.shape, "k_pe.shape", k_pe.shape)
         key_states = paddle.empty([bsz, q_len, self.num_heads, self.q_head_dim], dtype=self.config.dtype)
-        print("key_states.shape:", key_states.shape)
         # input[0]'s shape = [1, 2048, 16, 128], input[1]'s shape = [1, 2048, 1, 64].
         key_states = paddle.concat([k_nope, k_pe.expand([bsz, q_len, self.num_heads, k_pe.shape[-1]])], axis=3)