fix bug in scheduler.step(). Indices should be either on cpu or on th…#11038

Open

LiaoFJ wants to merge 1 commit intohuggingface:mainfrom

LiaoFJ:fix-bug-scheduler-step()

Commits on Mar 12, 2025

fix bug in scheduler.step(). Indices should be either on cpu or on the same device as the indexed tensor. (same as func add_noise())
LiaoFJ
committed