From 04e6806e413fc880aa864e6fecb12e185cda0625 Mon Sep 17 00:00:00 2001
From: kijai <jukka.seppanen@poke.fi>
Date: Sat, 16 Sep 2023 21:07:52 +0300
Subject: [PATCH] Update nodes.py

Smoother pingpong loop
---
 animatediff/nodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/animatediff/nodes.py b/animatediff/nodes.py
index 1fa3823..7c8e0ea 100644
--- a/animatediff/nodes.py
+++ b/animatediff/nodes.py
@@ -471,7 +471,7 @@ class AnimateDiffCombine:
             img = Image.fromarray(np.clip(img, 0, 255).astype(np.uint8))
             pil_images.append(img)
         if pingpong:
-            pil_images += pil_images[::-1]
+            pil_images += pil_images[-2:0:-1]
             
         # save image
         output_dir = (
@@ -610,7 +610,7 @@ class AnimateDiffCombineVideo:
             img = Image.fromarray(np.clip(img, 0, 255).astype(np.uint8))
             frames.append(img)
         if pingpong:
-            frames += frames[::-1]
+            frames += frames[-2:0:-1]
 
         with subprocess.Popen(args, stdin=subprocess.PIPE) as proc:
             for frame in frames: