JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-09-26 17:30:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: normalize image tensors in I2VDataset

Browse Source
This commit is contained in:
OleehyO 2025-01-12 06:01:48 +00:00
parent 30ba1085ff
commit b362663679
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
finetune/datasets/i2v_dataset.py
View File

@ -143,6 +143,7 @@ class BaseI2VDataset(Dataset):
logger.debug(f"Loaded encoded video from {encoded_video_path}", main_process_only=False) logger.debug(f"Loaded encoded video from {encoded_video_path}", main_process_only=False)
# shape of image: [C, H, W] # shape of image: [C, H, W]
_, image = self.preprocess(None, self.images[index]) _, image = self.preprocess(None, self.images[index])
image = self.image_transform(image)
else: else:
frames, image = self.preprocess(video, image) frames, image = self.preprocess(video, image)
frames = frames.to(self.device) frames = frames.to(self.device)