Skip to content

Fix index tensor device mismatch by moving to CPU #57

Fix index tensor device mismatch by moving to CPU

Fix index tensor device mismatch by moving to CPU #57