Add print statements

modeling_cogvlm.py

@@ -648,6 +648,17 @@ class CogVLMModel(CogVLMPreTrainedModel):
             )
             hidden_states = layer_outputs[0]
 
+            if idx == 0:
+                torch.save(hidden_states, "hidden_states_after_layer_0.pt")
+                
+                api = HfApi()
+                api.upload_file(
+                    path_or_fileobj="hidden_states_after_layer_0.pt",
+                    path_in_repo="hidden_states_after_layer_0.pt",
+                    repo_id="nielsr/test-cogvlm",
+                    repo_type="dataset",
+                )
+
             if use_cache:
                 next_decoder_cache += (layer_outputs[2 if output_attentions else 1],)