Update handler.py

handler.py

@@ -62,9 +62,9 @@ class EndpointHandler:
         with torch.no_grad():
             frames = self.vae.decode(tensor, return_dict=False)[0]
 
-        frames = frames.movedim(2, -1)
-        frames = (frames * 0.5 + 0.5).clamp(0, 1)
-        frames = frames.contiguous().float()
+        frames = frames[0].permute(1, 0, 2, 3)
+        frames = torch.stack([(frame * 0.5 + 0.5).clamp(0, 1) for frame in frames])
+        frames = frames.permute(0, 2, 3, 1).contiguous().float()
         frames = (frames * 255).round().to(torch.uint8)
 
         return _tobytes(frames, "frames")