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