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

Add mps backend for Pytorch in HF models #138

Merged
merged 1 commit into from
Jun 7, 2023

Conversation

arunpatro
Copy link
Contributor

Add support for mps backend to optimize inference on Apple Silicon

@arunpatro arunpatro requested a review from rlouf June 7, 2023 07:03
@arunpatro arunpatro changed the title add mps backend for pytorch in HF Transformers models add mps backend for Pytorch in HF models Jun 7, 2023
@rlouf rlouf changed the title add mps backend for Pytorch in HF models Add mps backend for Pytorch in HF models Jun 7, 2023
@rlouf
Copy link
Member

rlouf commented Jun 7, 2023

Please follow these guidelines when writing commit messages, and also keep in mind that PR titles are used to fill release notes automatically. I made the changes myself for this time.

LGTM otherwise, merging.

@rlouf rlouf merged commit b4e7113 into outlines-dev:main Jun 7, 2023
3 checks passed
@arunpatro arunpatro deleted the hf_mps branch June 7, 2023 20:43
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.

None yet

2 participants