Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Sync token-classification pipeline with Hub spec #34064

Open
wants to merge 9 commits into
base: main
Choose a base branch
from

Conversation

Rocketknight1
Copy link
Member

@Rocketknight1 Rocketknight1 commented Oct 10, 2024

One unusual change this time: The TokenClassificationPipeline put the docstrings in an odd place, with a decorator. As far as I can tell, this isn't necessary at all - I just put them in the normal place (which also un-breaks my compatibility checker).

cc @Wauplin - there's an output discrepancy for the zero-shot pipeline, but I can't figure out what output format the Hub spec actually wants when multi-label is True. Where did that Hub spec come from?

@Rocketknight1 Rocketknight1 force-pushed the sync_token_classification_and_zero_shot branch from e911c78 to 73e256c Compare October 10, 2024 12:56
@Wauplin
Copy link
Contributor

Wauplin commented Oct 10, 2024

@Rocketknight1 these specs comes from https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/src/tasks/zero-shot-classification/spec/output.json (PR huggingface/huggingface.js#449). When we wrote them we tried to be as consistent as possible with the existing but if there are some discrepancies, we can also update the specs correctly. Some tasks are more popular and have been reviewed/reworked several times but the zero-shot classification is not one of them 😕

@Rocketknight1 Rocketknight1 force-pushed the sync_token_classification_and_zero_shot branch from ba1f33b to f8fa11e Compare October 18, 2024 13:48
@Rocketknight1 Rocketknight1 changed the title Sync token-classification and zero-shot pipelines with Hub spec Sync token-classification pipeline with Hub spec Oct 18, 2024
@Rocketknight1
Copy link
Member Author

(This PR is ready, just on hold until the next release of huggingface_hub - will retest and merge then!)

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