andrewqian123
/

LLAMA_BATCH

Visual Question Answering

Model card Files Files and versions

andrewqian123 commited on Sep 4, 2024

Commit

c02c994

·

verified ·

1 Parent(s): 4331010

Update modeling_minicpmv.py

Files changed (1) hide show

modeling_minicpmv.py +2 -2

modeling_minicpmv.py CHANGED Viewed

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

                 to_add.append(0)
             # 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