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

Adds default token_type_ids when running on cpu for ORTModelFeatureExtraction #1499

Conversation

mhattingpete
Copy link

1498, added default token_type_ids when running on cpu for ORTModelFeatureExtraction

What does this PR do?

This PR adds a default token_type_ids when running on cpu for ORTModelFeatureExtraction in the same manner as attention_mask is set.

Fixes #1498

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you make sure to update the documentation with your changes?
  • Did you write any new necessary tests?

@mhattingpete mhattingpete deleted the 1498_fix_missing_token_type_ids_on_cpu branch September 2, 2024 07:47
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.

Error when running Feature extraction model on CPU
1 participant