From 0dc56d9dcc8a1a17a6a577a9b346c90234a8d22a Mon Sep 17 00:00:00 2001 From: mohui37 <77723348+mohui37@users.noreply.github.com> Date: Fri, 11 Apr 2025 17:05:40 +0800 Subject: [PATCH] Update train_wan_t2v.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 在应用itv的管道处理数据时有bug，提交修复 --- examples/wanvideo/train_wan_t2v.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/wanvideo/train_wan_t2v.py b/examples/wanvideo/train_wan_t2v.py index 8e8b370..ca199a2 100644 --- a/examples/wanvideo/train_wan_t2v.py +++ b/examples/wanvideo/train_wan_t2v.py @@ -140,7 +140,7 @@ class LightningModelForDataProcess(pl.LightningModule): if "first_frame" in batch: first_frame = Image.fromarray(batch["first_frame"][0].cpu().numpy()) _, _, num_frames, height, width = video.shape - image_emb = self.pipe.encode_image(first_frame, num_frames, height, width) + image_emb = self.pipe.encode_image(first_frame, None, num_frames, height, width) else: image_emb = {} data = {"latents": latents, "prompt_emb": prompt_emb, "image_emb": image_emb}