From 7303e37457d39ec841317292c2294f9c806039f0 Mon Sep 17 00:00:00 2001 From: kosch Date: Fri, 27 Oct 2023 11:01:12 +0800 Subject: [PATCH] a bug in line 113 --- TensorRT/demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TensorRT/demo.py b/TensorRT/demo.py index 456fad0..c856ea4 100644 --- a/TensorRT/demo.py +++ b/TensorRT/demo.py @@ -110,7 +110,7 @@ def predict( ) # compute the previous noisy sample x_t -> x_t-1 - latents = self.scheduler.step(noise_pred.cuda(), t, latents)[ + latents = self.scheduler.step(noise_pred.cuda(), i, latents)[ "prev_sample" ]