From 490d420d822472fc556ff01e96b9c4edd1eb953e Mon Sep 17 00:00:00 2001
From: wang96 <1290220814@qq.com>
Date: Thu, 27 Feb 2025 15:26:39 +0800
Subject: [PATCH] fix bugs

---
 examples/wanvideo/train_wan_i2v.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/wanvideo/train_wan_i2v.py b/examples/wanvideo/train_wan_i2v.py
index f10ff9f..9a49b34 100644
--- a/examples/wanvideo/train_wan_i2v.py
+++ b/examples/wanvideo/train_wan_i2v.py
@@ -24,8 +24,8 @@ class I2VDataset(torch.utils.data.Dataset):
         self.width = width
             
         self.frame_process = v2.Compose([
-            v2.Resize(size=(height, width), antialias=True),
             v2.CenterCrop(size=(height, width)),
+            v2.Resize(size=(height, width), antialias=True),
             v2.ToTensor(),
             v2.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]),
         ])
@@ -80,7 +80,7 @@ class I2VDataset(torch.utils.data.Dataset):
 
 
     def __getitem__(self, data_id):
-        text = self.path[data_id]
+        text = self.text[data_id]
         path = self.path[data_id]
         video, first_frame_image = self.load_video(path)
         data = {"text": text, "video": video, "first_frame_image":np.array(first_frame_image), "path": path}