Add print statements

modeling_cogvlm.py

@@ -241,6 +241,10 @@ class VisionExpertAttention(nn.Module):
         key_states = self._transpose_for_scores(key_states)  # B, H, L, HD
         value_states = self._transpose_for_scores(value_states)  # B, H, L, HD
 
+        if print_values:
+            print("Shape of query_states:", query_states.shape)
+            print("First values of query_states:", query_states[0,0,:3,:3])
+
         # if print_values:
 
             # torch.save(query_states, "query_states.pt")