nielsr HF staff commited on
Commit
419e827
1 Parent(s): 96b9145

Add print statements

Browse files
Files changed (1) hide show
  1. modeling_cogvlm.py +2 -0
modeling_cogvlm.py CHANGED
@@ -559,6 +559,8 @@ class CogVLMModel(CogVLMPreTrainedModel):
559
 
560
  torch.save(hidden_states, "inputs_embeds.pt")
561
 
 
 
562
  api = HfApi()
563
  api.upload_file(
564
  path_or_fileobj="inputs_embeds.pt",
 
559
 
560
  torch.save(hidden_states, "inputs_embeds.pt")
561
 
562
+ from huggingface_hub import HfApi
563
+
564
  api = HfApi()
565
  api.upload_file(
566
  path_or_fileobj="inputs_embeds.pt",