Update modeling_internvl_chat.py
Browse files
modeling_internvl_chat.py
CHANGED
|
@@ -320,7 +320,7 @@ class InternVLChatModel(PreTrainedModel):
|
|
| 320 |
vit_embeds = visual_features
|
| 321 |
else:
|
| 322 |
vit_embeds = []
|
| 323 |
-
num_chunks =
|
| 324 |
pixel_values_splitted = pixel_values.chunk(num_chunks)
|
| 325 |
|
| 326 |
for pixel_values_ in pixel_values_splitted:
|
|
|
|
| 320 |
vit_embeds = visual_features
|
| 321 |
else:
|
| 322 |
vit_embeds = []
|
| 323 |
+
num_chunks = 1
|
| 324 |
pixel_values_splitted = pixel_values.chunk(num_chunks)
|
| 325 |
|
| 326 |
for pixel_values_ in pixel_values_splitted:
|