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

Quick fix for XUMX in torch 2.0 #684

Merged
merged 1 commit into from
Mar 29, 2024

Conversation

DavidDiazGuerra
Copy link
Contributor

Hello!

I know we should ideally rewrite the XUMX model to work with the new torch's complex type, but it is possible to workaround this by just converting the new type to the old format just after calling torch.stft using torch.view_as_real and then converting it back to the new type with torch.view_as_complex before calling to torch.istft (#662) . This is not the ideal solution but I think it would be useful to allow people to keep using XUMX till the model is fully rewrited.

Best,
David

@mpariente
Copy link
Collaborator

Thank you very much for the PR !

@r-sawata, do you agree ?

@r-sawata
Copy link
Contributor

Yea I confirmed this PR and actually tried to run it on my local. It looks having no any problems and will be suitable for future update of PyTorch, so I totally agree with this PR.

Thank you so much for your PR, @DavidDiazGuerra.

Could you merge this to master if you're fine, @mpariente?

@mpariente mpariente merged commit d10b407 into asteroid-team:master Mar 29, 2024
5 of 6 checks passed
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.

3 participants