fix encode caching bug

TNTwise · Sep 14, 2024 · 63fcb01 · 63fcb01
1 parent cd047eb
commit 63fcb01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/src/InterpolateTorch.py b/backend/src/InterpolateTorch.py
@@ -355,7 +355,7 @@ def process(self, img0, img1, timestep):
             timestep = self.timestepDict[timestep]
             if not self.rife46:
                 if self.f1encode is None:
-                    self.f1encode = self.encode(img0[:, :3])
+                    self.f1encode = self.encode(img1[:, :3])
                 output, self.f1encode = self.flownet(
                     img0, img1, timestep, self.f1encode
                 )