Add print statements

modeling_cogvlm.py

@@ -439,7 +439,7 @@ class CogVLMModel(CogVLMPreTrainedModel):
                 images_features = images_features.to(dtype=inputs_embeds.dtype, device=inputs_embeds.device)
 
                 print("First values of text embeddings:", inputs_embeds[0, :3, :3])
-                print("First values of images_features:", images_features[0, :3, :3])
+                print("First values of images_features:", images_features[0, :3])
 
                 inputs_embeds = inputs_embeds.index_put([token_type_ids == VISION_TOKEN_TYPE], images_features)