Update handler.py

handler.py

@@ -49,12 +49,12 @@ class EndpointHandler:
             latents_mean = (
                 torch.tensor(self.vae.config.latents_mean)
                 .view(1, self.vae.config.z_dim, 1, 1, 1)
-                .to(latents.device, latents.dtype)
+                .to(tensor.device, tensor.dtype)
             )
             latents_std = 1.0 / torch.tensor(self.vae.config.latents_std).view(
                 1, self.vae.config.z_dim, 1, 1, 1
-            ).to(latents.device, latents.dtype)
-            latents = latents / latents_std + latents_mean
+            ).to(tensor.device, tensor.dtype)
+            tensor = tensor / latents_std + latents_mean
 
         with torch.no_grad():
             frames = cast(torch.Tensor, self.vae.decode(tensor, return_dict=False)[0])