`unsupported datatype for constant'

mps
1

hi, I’m trying to test this code:

https://pytorch-lightning.readthedocs.io/en/stable/notebooks/course_UvA-DL/13-contrastive-learning.html

with mps device on M1 with torch version 1.13.0.dev20220913 but I get

/AppleInternal/Library/BuildRoots/a0876c02-1788-11ed-b9c4-96898e02b808/Library/Caches/com.apple.xbs/Sources/MetalPerformanceShadersGraph/mpsgraph/MetalPerformanceShadersGraph/Core/Files/MPSGraphUtilities.mm:343: failed assertion 'unsupported datatype for constant'

I have confirmed that mps device is available and works for other code