Skip to content

Add mlx support to BatchEncoding.convert_to_tensors#29406

Merged
ArthurZucker merged 4 commits intohuggingface:mainfrom
tidely:main
Mar 4, 2024
Merged

Add mlx support to BatchEncoding.convert_to_tensors#29406
ArthurZucker merged 4 commits intohuggingface:mainfrom
tidely:main

Conversation

@tidely
Copy link
Contributor

@tidely tidely commented Mar 2, 2024

What does this PR do?

Adds support for the mlx array framework by Apple to BatchEncoding.convert_to_tensors

Fixes #29405

@tidely tidely marked this pull request as ready for review March 2, 2024 15:37
@tidely
Copy link
Contributor Author

tidely commented Mar 2, 2024

Should be ready for review @ArthurZucker

Copy link
Collaborator

@ArthurZucker ArthurZucker left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, clean addition thanks

@ArthurZucker
Copy link
Collaborator

Failing test is unrelated

@ArthurZucker ArthurZucker merged commit 704b3f7 into huggingface:main Mar 4, 2024
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.

Add mlx support for BatchEncoding.convert_to_tensors

2 participants