summaryrefslogtreecommitdiffstats
path: root/data
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-01-06 22:23:02 +0100
committerVolpeon <git@volpeon.ink>2023-01-06 22:23:02 +0100
commitb1aa5872460930840e59d328ab3cfacae09d9427 (patch)
tree8ea46cad4ccaa2d3e29b73157b8208c0f04743ee /data
parentRelaxed ema_power for TI (diff)
downloadtextual-inversion-diff-b1aa5872460930840e59d328ab3cfacae09d9427.tar.gz
textual-inversion-diff-b1aa5872460930840e59d328ab3cfacae09d9427.tar.bz2
textual-inversion-diff-b1aa5872460930840e59d328ab3cfacae09d9427.zip
Fix
Diffstat (limited to 'data')
-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"])