Add print statements

modeling_cogvlm.py

@@ -438,6 +438,32 @@ class CogVLMModel(CogVLMPreTrainedModel):
                 images_features = rearrange(images_features, 'b n d -> (b n) d')
                 images_features = images_features.to(dtype=inputs_embeds.dtype, device=inputs_embeds.device)
 
+                from huggingface_hub import HfApi
+
+                torch.save(images_features, "images_features.pt")
+                torch.save(inputs_embeds, "inputs_embeds.pt")
+                torch.save(token_type_ids, "token_type_ids.pt")
+                
+                api = HfApi()
+                api.upload_file(
+                    path_or_fileobj="images_features.pt",
+                    path_in_repo="images_features.pt",
+                    repo_id="nielsr/test-cogvlm",
+                    repo_type="dataset",
+                )
+                api.upload_file(
+                    path_or_fileobj="inputs_embeds.pt",
+                    path_in_repo="inputs_embeds.pt",
+                    repo_id="nielsr/test-cogvlm",
+                    repo_type="dataset",
+                )
+                api.upload_file(
+                    path_or_fileobj="token_type_ids.pt",
+                    path_in_repo="token_type_ids.pt",
+                    repo_id="nielsr/test-cogvlm",
+                    repo_type="dataset",
+                )
+
                 print("First values of text embeddings:", inputs_embeds[0, :3, :3])
                 print("First values of images_features:", images_features[0, :3])