Skip to content

Conversation

@Adel-Moumen
Copy link
Collaborator

Small fix for the mac MPS error #1794

It works perfectly as expected on my mac m1 with this fix.

Please note that MPS is not yet supported for all operations, and in #1794 it will be necessary to run the script with PYTORCH_ENABLE_MPS_FALLBACK=1

Copy link
Collaborator

@pplantinga pplantinga left a comment

Choose a reason for hiding this comment

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

Nice fix! LGTM

Its curious that these are the only two instances of this kind of code. Makes me wonder if its necessary at all or if there's another way it could be done. E.g. the input is always floating point so .to(input.device) is all that is needed.

@anautsch anautsch merged commit 0423bda into speechbrain:develop Jan 17, 2023
@mattiasu96
Copy link

When this will be released? Otherwise im gonna just change my local code 🤔

@Adel-Moumen
Copy link
Collaborator Author

@mattiasu96 please do :

git clone https://github.com/speechbrain/speechbrain.git
cd speechbrain
pip install -r requirements.txt
pip install --editable .

It will download and install the new SpeechBrain version with the fix.

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.

4 participants