Add print statements

modeling_cogvlm.py

@@ -598,6 +598,7 @@ class CogVLMModel(CogVLMPreTrainedModel):
             torch.save(attention_mask, "attention_mask_step_1.pt")
             torch.save(token_type_ids, "token_type_ids_step_1.pt")
             torch.save(position_ids, "position_ids_step_1.pt")
+            torch.save(past_key_values, "past_key_value_step_1.pt")
             
             from huggingface_hub import HfApi