From 2d4cbba16485ac60e92bb879ad62383e131d3448 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Tue, 17 Dec 2024 15:26:01 +0100 Subject: [PATCH] temp --- tests/models/kosmos2_5/test_modeling_kosmos2_5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/kosmos2_5/test_modeling_kosmos2_5.py b/tests/models/kosmos2_5/test_modeling_kosmos2_5.py index bee586b42..9a28b3eb5 100644 --- a/tests/models/kosmos2_5/test_modeling_kosmos2_5.py +++ b/tests/models/kosmos2_5/test_modeling_kosmos2_5.py @@ -255,7 +255,7 @@ class Kosmos2_5ModelTester: ): model = Kosmos2_5Model(config).to(torch_device).eval() with torch.no_grad(): - result = model(flattened_patches, input_ids, image_embeds_position_mask, attention_mask) + result = model(input_ids, flattened_patches, image_embeds_position_mask, attention_mask) self.parent.assertEqual( result.last_hidden_state.shape, (