Update examples/dreambooth/train_dreambooth_lora_flux.py

davidberenstein1957 · web-flow · commit b9e1ccf0b499 · 2024-12-11T10:04:16.000+01:00
diff --git a/examples/dreambooth/train_dreambooth_lora_flux.py b/examples/dreambooth/train_dreambooth_lora_flux.py
@@ -742,6 +742,7 @@ def __init__(
                     raise ValueError(
                         f"`--image_column` value '{args.image_column}' not found in dataset columns. Dataset columns are: {', '.join(column_names)}"
                     )
+            # force casting for an image column to decode when this isn't the default
             dataset["train"] = dataset["train"].cast_column(image_column, Image(decode=True))
             instance_images = dataset["train"][image_column]
 

Original file line number	Diff line number	Diff line change
`@@ -742,6 +742,7 @@ def __init__(`
`742`	`742`	`raise ValueError(`
`743`	`743`	f"`--image_column` value '{args.image_column}' not found in dataset columns. Dataset columns are: {', '.join(column_names)}"
`744`	`744`	`)`
	`745`	`+ # force casting for an image column to decode when this isn't the default`
`745`	`746`	`dataset["train"] = dataset["train"].cast_column(image_column, Image(decode=True))`
`746`	`747`	`instance_images = dataset["train"][image_column]`
`747`	`748`