• brkirch's avatar
    MPS fix is still needed :( · 7738c057
    brkirch authored
    Apparently I did not test with large enough images to trigger the bug with torch.narrow on MPS
    7738c057
devices.py 6.96 KB