fix paddle zero output for release3.0b2 and develop (#1244)

pkhk-1 · web-flow · commit 21aae876c4fd · 2025-04-24T15:10:54.000+08:00
diff --git a/paddlemix/models/llava/llava_arch.py b/paddlemix/models/llava/llava_arch.py
@@ -344,7 +344,7 @@ def prepare_inputs_labels_for_multimodal(
                 continue
             image_token_indices = (
                 [-1]
-                + paddle.where(cur_input_ids == IMAGE_TOKEN_INDEX)[0].tolist()
+                + paddle.squeeze(paddle.where(cur_input_ids == IMAGE_TOKEN_INDEX)).unsqueeze(-1).tolist()
                 + [cur_input_ids.shape[0]]
             )
 

Original file line number	Diff line number	Diff line change
`@@ -344,7 +344,7 @@ def prepare_inputs_labels_for_multimodal(`
`344`	`344`	`continue`
`345`	`345`	`image_token_indices = (`
`346`	`346`	`[-1]`
`347`		`- + paddle.where(cur_input_ids == IMAGE_TOKEN_INDEX)[0].tolist()`
	`347`	`+ + paddle.squeeze(paddle.where(cur_input_ids == IMAGE_TOKEN_INDEX)).unsqueeze(-1).tolist()`
`348`	`348`	`+ [cur_input_ids.shape[0]]`
`349`	`349`	`)`
`350`	`350`