andrewqian123
/

LLAMA_BATCH

Visual Question Answering

Model card Files Files and versions

andrewqian123 commited on Sep 6, 2024

Commit

21f37b3

·

verified ·

1 Parent(s): 43a4dd1

Update modeling_minicpmv.py

Files changed (1) hide show

modeling_minicpmv.py +4 -4

modeling_minicpmv.py CHANGED Viewed

@@ -310,10 +310,10 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
                 tensor = torch.cat((vector_reshaped, tensor), dim=1)
                 to_add.append(1)
             # tensor = tensor.to(self.device)
-            if bound != 0:
-                print(tensor[0,bound-1], " BOUND")
-                print(tensor[0,bound], " OTHERSIDE")
-                print(pad_embedding_vector, " VECTOR")
             print(tensor.shape, "UPDATED_SHAPE")
             batch[place] = tensor
             attention_mask.append(to_add)

                 tensor = torch.cat((vector_reshaped, tensor), dim=1)
                 to_add.append(1)
             # tensor = tensor.to(self.device)
+            # if bound != 0:
+            #     print(tensor[0,bound-1], " BOUND")
+            #     print(tensor[0,bound], " OTHERSIDE")
+            #     print(pad_embedding_vector, " VECTOR")
             print(tensor.shape, "UPDATED_SHAPE")
             batch[place] = tensor
             attention_mask.append(to_add)