Add print statements

modeling_cogvlm.py

@@ -556,6 +556,9 @@ class CogVLMModel(CogVLMPreTrainedModel):
         hidden_states = inputs_embeds
 
         torch.save(hidden_states, "initial_hidden_states.pt")
+        torch.save(attention_mask, "initial_attention_mask.pt")
+        torch.save(token_type_ids, "initial_token_type_ids.pt")
+        torch.save(position_ids, "initial_position_ids.pt")
         
         from huggingface_hub import HfApi
 
@@ -566,6 +569,27 @@ class CogVLMModel(CogVLMPreTrainedModel):
             repo_id="nielsr/test-cogvlm",
             repo_type="dataset",
         )
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj="initial_attention_mask.pt",
+            path_in_repo="initial_attention_mask.pt",
+            repo_id="nielsr/test-cogvlm",
+            repo_type="dataset",
+        )
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj="initial_token_type_ids.pt",
+            path_in_repo="initial_token_type_ids.pt",
+            repo_id="nielsr/test-cogvlm",
+            repo_type="dataset",
+        )
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj="initial_position_ids.pt",
+            path_in_repo="initial_position_ids.pt",
+            repo_id="nielsr/test-cogvlm",
+            repo_type="dataset",
+        )
 
         # decoder layers
         all_hidden_states = () if output_hidden_states else None