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.

Sign up for GitHub

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

Jump to bottom

[FEAT] Support Metal and Triton #1

Open
mikejgray opened this issue Sep 29, 2024 · 0 comments
Open

[FEAT] Support Metal and Triton #1

mikejgray opened this issue Sep 29, 2024 · 0 comments
Labels
help wanted Extra attention is needed

Comments

@mikejgray
Copy link

Transformers will allow you to pass in support for Metal (Apple Silicon) and Triton (AMD) GPUs. It would be nice to have the ability to pass in which GPU you want to use for inference as a config option, defaulting back to CPU if it fails
@JarbasAl JarbasAl added the help wanted Extra attention is needed label Sep 29, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
help wanted Extra attention is needed
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@mikejgray @JarbasAl