diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-06 22:23:02 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-06 22:23:02 +0100 |
| commit | b1aa5872460930840e59d328ab3cfacae09d9427 (patch) | |
| tree | 8ea46cad4ccaa2d3e29b73157b8208c0f04743ee /data | |
| parent | Relaxed ema_power for TI (diff) | |
| download | textual-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.py | 6 |
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"]) |