nielsr HF staff commited on
Commit
77a0d59
1 Parent(s): 4bacabb

Add print statements

Browse files
Files changed (1) hide show
  1. modeling_cogvlm.py +1 -0
modeling_cogvlm.py CHANGED
@@ -598,6 +598,7 @@ class CogVLMModel(CogVLMPreTrainedModel):
598
  torch.save(attention_mask, "attention_mask_step_1.pt")
599
  torch.save(token_type_ids, "token_type_ids_step_1.pt")
600
  torch.save(position_ids, "position_ids_step_1.pt")
 
601
 
602
  from huggingface_hub import HfApi
603
 
 
598
  torch.save(attention_mask, "attention_mask_step_1.pt")
599
  torch.save(token_type_ids, "token_type_ids_step_1.pt")
600
  torch.save(position_ids, "position_ids_step_1.pt")
601
+ torch.save(past_key_values, "past_key_value_step_1.pt")
602
 
603
  from huggingface_hub import HfApi
604