Add print statements

modeling_cogvlm.py

@@ -602,33 +602,37 @@ class CogVLMModel(CogVLMPreTrainedModel):
             from huggingface_hub import HfApi
 
             api = HfApi()
+
             api.upload_file(
                 path_or_fileobj="hidden_states_step_1.pt",
                 path_in_repo="hidden_states_step_1.pt",
                 repo_id="nielsr/test-cogvlm",
                 repo_type="dataset",
             )
-            api = HfApi()
             api.upload_file(
                 path_or_fileobj="attention_mask_step_1.pt",
                 path_in_repo="attention_mask_step_1.pt",
                 repo_id="nielsr/test-cogvlm",
                 repo_type="dataset",
             )
-            api = HfApi()
             api.upload_file(
                 path_or_fileobj="token_type_ids_step_1.pt",
                 path_in_repo="token_type_ids_step_1.pt",
                 repo_id="nielsr/test-cogvlm",
                 repo_type="dataset",
             )
-            api = HfApi()
             api.upload_file(
                 path_or_fileobj="position_ids_step_1.pt",
                 path_in_repo="position_ids_step_1.pt",
                 repo_id="nielsr/test-cogvlm",
                 repo_type="dataset",
             )
+            api.upload_file(
+                path_or_fileobj="past_key_value_step_1.pt",
+                path_in_repo="past_key_value_step_1.pt",
+                repo_id="nielsr/test-cogvlm",
+                repo_type="dataset",
+            )
 
         # decoder layers
         all_hidden_states = () if output_hidden_states else None