Skip to content

Commit c9fedf1

Browse files
Apply suggestions from code review
1 parent 2c1cbb3 commit c9fedf1

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

examples/dreambooth/train_dreambooth_lora_flux.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -742,7 +742,6 @@ def __init__(
742742
raise ValueError(
743743
f"`--image_column` value '{args.image_column}' not found in dataset columns. Dataset columns are: {', '.join(column_names)}"
744744
)
745-
746745
dataset["train"] = dataset["train"].cast_column(image_column, Image(decode=True))
747746
instance_images = dataset["train"][image_column]
748747

@@ -770,7 +769,6 @@ def __init__(
770769

771770
instance_images = [Image.open(path) for path in list(Path(instance_data_root).iterdir())]
772771
self.custom_instance_prompts = None
773-
774772
self.instance_images = []
775773
for img in instance_images:
776774
self.instance_images.extend(itertools.repeat(img, repeats))
@@ -786,7 +784,6 @@ def __init__(
786784
]
787785
)
788786
for image in self.instance_images:
789-
790787
image = exif_transpose(image)
791788
if not image.mode == "RGB":
792789
image = image.convert("RGB")

0 commit comments

Comments
 (0)