Update modeling_minicpmv.py
Browse files- modeling_minicpmv.py +1 -0
modeling_minicpmv.py
CHANGED
|
@@ -303,6 +303,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
|
|
| 303 |
for stuff in batch:
|
| 304 |
print(stuff.shape, "SHAPE")
|
| 305 |
batch = torch.cat(padded_tensors, dim=0)
|
|
|
|
| 306 |
|
| 307 |
# output_ids = self._decode(input_embeds, tokenizer, **kwargs)
|
| 308 |
if stream:
|
|
|
|
| 303 |
for stuff in batch:
|
| 304 |
print(stuff.shape, "SHAPE")
|
| 305 |
batch = torch.cat(padded_tensors, dim=0)
|
| 306 |
+
print(batch.shape)
|
| 307 |
|
| 308 |
# output_ids = self._decode(input_embeds, tokenizer, **kwargs)
|
| 309 |
if stream:
|