diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-03-24 03:04:47 -0400 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-03-24 04:04:22 -0400 |
| commit | 27fe3eb6a9d8f866af8b90dff18f4445124702da (patch) | |
| tree | 1ba1eb23105a690cec2e4c8dd391b82a169ccd8b /scripts/loopback.py | |
| parent | c5142e2fbecb50531a55aa804ea132c5d870858c (diff) | |
Add workaround for MPS layer_norm on PyTorch 2.0
On PyTorch 2.0, with MPS layer_norm only accepts float32 inputs. This was fixed shortly after 2.0 was finalized so the workaround can be applied with an exact version match.
Diffstat (limited to 'scripts/loopback.py')
0 files changed, 0 insertions, 0 deletions