Skip to content

Account for additional processor outputs#4191

Closed
KarelKenens wants to merge 4 commits intohuggingface:mainfrom
KarelKenens:account-for-additional-processor-outputs
Closed

Account for additional processor outputs#4191
KarelKenens wants to merge 4 commits intohuggingface:mainfrom
KarelKenens:account-for-additional-processor-outputs

Conversation

@KarelKenens
Copy link

What does this PR do?

In DataCollatorForVisionLanguageModeling, attempt to dynamically determine additional processor outputs, on top of input_ids and attention_mask. For example in the Gemma 3 model family also token_type_ids are part of the output (and required futher downstream).

Fixes #4189

@qgallouedec , similar to #4190

@qgallouedec
Copy link
Member

closing in favour of #4190, thanks @KarelKenens

@qgallouedec qgallouedec closed this Oct 9, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

sft_trainer.DataCollatorForVisionLanguageModelling does not account for "non-standard" processor outputs (e.g. Gemma 3)

2 participants

Comments