summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--data/csv.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/data/csv.py b/data/csv.py
index d1f3054..4986153 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -246,12 +246,14 @@ class CSVDataset(Dataset):
246 246
247 example = {} 247 example = {}
248 248
249 example["prompts"] = keywords_to_prompt(unprocessed_example["prompts"], self.dropout, True) 249 example["prompts"] = keywords_to_prompt(unprocessed_example["prompts"])
250 example["cprompts"] = unprocessed_example["cprompts"] 250 example["cprompts"] = unprocessed_example["cprompts"]
251 example["nprompts"] = unprocessed_example["nprompts"] 251 example["nprompts"] = unprocessed_example["nprompts"]
252 252
253 example["instance_images"] = self.image_transforms(unprocessed_example["instance_images"]) 253 example["instance_images"] = self.image_transforms(unprocessed_example["instance_images"])
254 example["instance_prompt_ids"] = self.prompt_processor.get_input_ids(example["prompts"]) 254 example["instance_prompt_ids"] = self.prompt_processor.get_input_ids(
255 keywords_to_prompt(unprocessed_example["prompts"], self.dropout, True)
256 )
255 257
256 if self.num_class_images != 0: 258 if self.num_class_images != 0:
257 example["class_images"] = self.image_transforms(unprocessed_example["class_images"]) 259 example["class_images"] = self.image_transforms(unprocessed_example["class_images"])