saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

temp

Browse source
This commit is contained in:
ydshieh 2024-12-17 15:26:01 +01:00
parent 83d600e5f8
commit 2d4cbba164
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/models/kosmos2_5/test_modeling_kosmos2_5.py
View file

@ -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,
(