Increase minimum pytorch version and a couple quick fixes

Neutone / neutone_sdk

Join the community on Discord for more discussions around Neutone! https://discord.gg/VHSMzb8Wqp

GNU Lesser General Public License v2.1

465 stars 21 forks source link

Increase minimum pytorch version and a couple quick fixes #50

Closed bogdanteleaga closed 12 months ago

bogdanteleaga commented 1 year ago

The main goal of the PR is to increase the minimum pytorch version to support 2.0.x.

However, I ran into this bug https://github.com/pytorch/pytorch/issues/45208 so made a quick fix for it, and slightly increased the tolerance on the allclose check we do on exports. The diff in audio.py looks a bit off, but I just added a torch.inference_mode() call at the top.

christhetree commented 12 months ago

Thanks for finding that no_grad decorator info, LGTM