Skip to content

Conversation

@kharaldsson
Copy link
Collaborator

@kharaldsson kharaldsson commented Jan 30, 2025

  • Creates a conda environment sufficient to support training using mps compute on an Apple M[>=1] machine
  • Updates the names of existing conda environments to reflect cuda vs mps requirements

Note: While pytorch recently added support for mixed precision training using an mps backend, huggingface has yet to update the accelerate library accordingly (see, I think, accelerate autocast on mps device #1396)

@kharaldsson kharaldsson self-assigned this Jan 30, 2025
@kharaldsson kharaldsson requested a review from shanest February 14, 2025 16:55
@kharaldsson kharaldsson marked this pull request as ready for review February 14, 2025 16:55
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