Add print statements

modeling_cogvlm.py

@@ -532,6 +532,7 @@ class CogVLMModel(CogVLMPreTrainedModel):
 
             if idx in [0, 1, 2]:
                 print(f"Hidden states before layer {idx}", hidden_states[0,:3,:3])
+                print(f"Mean of hidden states before layer {idx}", hidden_states.mean())
 
             past_key_value = past_key_values[idx] if past_key_values is not None else None
             layer_outputs = decoder_layer(