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:28:45 +01:00
parent 2d4cbba164
commit 6b2f7d79e2
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -638,14 +638,14 @@ class Kosmos2_5ModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTester
main_input = inputs[main_input_name]
model(
main_input,
inputs["input_ids"],
inputs["flattened_patches"],
inputs["image_embeds_position_mask"],
)
traced_model = torch.jit.trace(
model,
(
main_input,
inputs["input_ids"],
inputs["flattened_patches"],
inputs["image_embeds_position_mask"],
),
)