padding fix

2024-11-11 17:29:57 +02:00
parent 43bc7fb4fc
commit 5f1a917b93
3 changed files with 23 additions and 9 deletions
@@ -1004,7 +1004,7 @@ class CogVideoDecode:
            vae._clear_fake_context_parallel_cache()
        except:
            pass
-        frames = vae.decode(latents).sample
+        frames = vae.decode(latents[:, :, pipeline["pipe"].additional_frames:]).sample
        vae.disable_tiling()
        if not pipeline["cpu_offloading"]:
            vae.to(offload_device)