fix order of running pil image dataloader plugin

2023-12-20 15:24:09 -05:00 · 2023-12-20 15:24:09 -05:00 · d9d2a43804
parent dfcc9e7f41
commit d9d2a43804
1 changed files with 1 additions and 1 deletions
--- a/data/every_dream.py
+++ b/data/every_dream.py
@ -102,8 +102,8 @@ class EveryDreamBatch(Dataset):
        else:
            example["caption"] = train_item["caption"].get_caption()

-        example["image"] = image_transforms(train_item["image"])
        example["image"] = self.plugin_runner.transform_pil_image(example["image"])
+        example["image"] = image_transforms(train_item["image"])        
        example["caption"] = self.plugin_runner.transform_caption(example["caption"])

        if random.random() > (train_item.get("cond_dropout", self.conditional_dropout)):