From 2871f9b9d538e24d8b4375076edfef4a5e905da7 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Thu, 28 Nov 2024 18:20:09 +0200 Subject: [PATCH] Update nodes.py --- nodes.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/nodes.py b/nodes.py index d84854f..feade09 100644 --- a/nodes.py +++ b/nodes.py @@ -301,15 +301,15 @@ class CogVideoConcatLatent: def encode(self, samples_from, samples_to): - insert_from = samples_from["samples"] - insert_to = samples_to["samples"] + insert_from = samples_from["samples"].clone() + insert_to = samples_to["samples"].clone() new_latents = torch.cat((insert_to, insert_from), dim=1) - print("new latents shape: ", new_latents.shape) - - - samples_to.update({"samples": new_latents}) - return (samples_to, ) + return ({ + "samples": new_latents, + "start_percent": samples_from["start_percent"], + "end_percent": samples_from["end_percent"] + }, ) class CogVideoImageEncodeFunInP: @classmethod