THUDM
/

cogvlm-chat-hf

Text Generation

Transformers

Safetensors

English

custom_code

Model card Files Files and versions Community

nielsr HF staff commited on Mar 25

Commit

d0bdef6

•

1 Parent(s): 419e827

Add print statements

Browse files

Files changed (1) hide show

modeling_cogvlm.py +25 -25

modeling_cogvlm.py CHANGED Viewed

@@ -440,29 +440,29 @@ class CogVLMModel(CogVLMPreTrainedModel):
                 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])
@@ -557,14 +557,14 @@ class CogVLMModel(CogVLMPreTrainedModel):
         hidden_states = inputs_embeds
-        torch.save(hidden_states, "inputs_embeds.pt")
         from huggingface_hub import HfApi
         api = HfApi()
         api.upload_file(
-            path_or_fileobj="inputs_embeds.pt",
-            path_in_repo="inputs_embeds.pt",
             repo_id="nielsr/test-cogvlm",
             repo_type="dataset",
         )

                 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])
         hidden_states = inputs_embeds
+        torch.save(hidden_states, "initial_hidden_states.pt")
         from huggingface_hub import HfApi
         api = HfApi()
         api.upload_file(
+            path_or_fileobj="initial_hidden_states.pt",
+            path_in_repo="initial_hidden_states.pt",
             repo_id="nielsr/test-cogvlm",
             repo_type="dataset",
         )