THUDM
/

cogvlm-chat-hf

Text Generation

Model card Files Files and versions Community

nielsr HF staff commited on Mar 12

Commit

9648b08

•

1 Parent(s): 87b061a

Add print statements

Files changed (1) hide show

modeling_cogvlm.py +1 -1

modeling_cogvlm.py CHANGED Viewed

@@ -439,7 +439,7 @@ class CogVLMModel(CogVLMPreTrainedModel):
                 images_features = images_features.to(dtype=inputs_embeds.dtype, device=inputs_embeds.device)
                 print("First values of text embeddings:", inputs_embeds[0, :3, :3])
-                print("First values of images_features:", images_features[0, :3, :3])
                 inputs_embeds = inputs_embeds.index_put([token_type_ids == VISION_TOKEN_TYPE], images_features)

                 images_features = images_features.to(dtype=inputs_embeds.dtype, device=inputs_embeds.device)
                 print("First values of text embeddings:", inputs_embeds[0, :3, :3])
+                print("First values of images_features:", images_features[0, :3])
                 inputs_embeds = inputs_embeds.index_put([token_type_ids == VISION_TOKEN_TYPE], images_features)