a bug in /Tensorrt/demo.py

[kx-kexi]

latents = self.scheduler.step(noise_pred.cuda(), t, latents)[
"prev_sample"
]
i think there is a bug in this code, it should be:
latents = self.scheduler.step(noise_pred.cuda(), i, latents)[
"prev_sample"
]
because if i use t as the input , it will occur IndexError: index 999 is out of bounds for dimension 0 with size 51
and i change t to i ,the code can run right