Update modeling_minicpmv.py
Browse files- modeling_minicpmv.py +1 -1
modeling_minicpmv.py
CHANGED
|
@@ -306,7 +306,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
|
|
| 306 |
bound = pl
|
| 307 |
|
| 308 |
#print("here")
|
| 309 |
-
tensor = torch.cat((
|
| 310 |
to_add.append(0)
|
| 311 |
# tensor = tensor.to(self.device)
|
| 312 |
if bound != 0:
|
|
|
|
| 306 |
bound = pl
|
| 307 |
|
| 308 |
#print("here")
|
| 309 |
+
tensor = torch.cat((vector_reshaped, tensor), dim=1)
|
| 310 |
to_add.append(0)
|
| 311 |
# tensor = tensor.to(self.device)
|
| 312 |
if bound != 0:
|