From 2f76bace4d2a2ebb429751ca245ef3c9a926d22e Mon Sep 17 00:00:00 2001 From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com> Date: Tue, 12 Nov 2024 10:28:13 +0800 Subject: [PATCH] Fix bug in comfyui t2v (#140) --- comfyui/comfyui_nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfyui/comfyui_nodes.py b/comfyui/comfyui_nodes.py index 136b313..0364d95 100644 --- a/comfyui/comfyui_nodes.py +++ b/comfyui/comfyui_nodes.py @@ -630,7 +630,6 @@ class EasyAnimateT2VSampler: clip_image = clip_image, comfyui_progressbar = True, ).videos - videos = rearrange(sample, "b c t h w -> (b t) h w c") else: sample = pipeline( prompt, @@ -642,6 +641,7 @@ class EasyAnimateT2VSampler: guidance_scale = cfg, num_inference_steps = steps, ).videos + videos = rearrange(sample, "b c t h w -> (b t) h w c") for _lora_path, _lora_weight in zip(easyanimate_model.get("loras", []), easyanimate_model.get("strength_model", [])): pipeline = unmerge_lora(pipeline, _lora_path, _lora_weight)