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 14:47:48 +01:00
parent 9ddc86b560
commit 39dc6ef257
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/transformers/models/kosmos2_5/modeling_kosmos2_5.py
View file

@ -1806,8 +1806,8 @@ class Kosmos2_5Model(Kosmos2_5PreTrainedModel):
@replace_return_docstrings(output_type=Kosmos2_5ModelOutput, config_class=_CONFIG_FOR_DOC)
def forward(
self,
flattened_patches: Optional[torch.Tensor] = None,
input_ids: Optional[torch.Tensor] = None,
flattened_patches: Optional[torch.Tensor] = None,
image_embeds_position_mask: Optional[torch.Tensor] = None,
attention_mask: Optional[torch.Tensor] = None,
past_key_values: Optional[Union[Cache, List[torch.FloatTensor]]] = None,