Skip to content

Conversation

@cceyda
Copy link
Contributor

@cceyda cceyda commented Aug 21, 2025

create_labels function returns a tuple labels_batch, decoder_tokenized_input

return labels_batch, decoder_tokenized_input

In SpanBiEncoderProcessor there is no need for label decoding stuff so we need to fix the returned value assignment. otherwise labels become tuple & it breaks stuff during forward.

@Ingvarstep
Copy link
Collaborator

Thank you for the fix.

@Ingvarstep Ingvarstep merged commit 10d8d29 into urchade:main Aug 21, 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.

2 participants