From 4f14db12694306a3fdf229c3394ddfaa5bbff000 Mon Sep 17 00:00:00 2001 From: Tianhao-Qi <37017794+Tianhao-Qi@users.noreply.github.com> Date: Fri, 18 Jul 2025 14:13:44 +0800 Subject: [PATCH] fix bug in getting mask_pixel_values for non-normal training settings (#252) --- videox_fun/data/dataset_image_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/videox_fun/data/dataset_image_video.py b/videox_fun/data/dataset_image_video.py index 8270aa7..3ea5b75 100755 --- a/videox_fun/data/dataset_image_video.py +++ b/videox_fun/data/dataset_image_video.py @@ -498,7 +498,7 @@ class ImageVideoDataset(Dataset): if self.enable_inpaint and not self.enable_bucket: mask = get_random_mask(pixel_values.size()) - mask_pixel_values = pixel_values * (1 - mask) + torch.ones_like(pixel_values) * -1 * mask + mask_pixel_values = pixel_values * (1 - mask) + torch.ones_like(pixel_values) * mask sample["mask_pixel_values"] = mask_pixel_values sample["mask"] = mask