diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py index 3df55937c4..2ecb8fb174 100644 --- a/examples/dreambooth/train_dreambooth.py +++ b/examples/dreambooth/train_dreambooth.py @@ -156,7 +156,10 @@ def parse_args(input_args=None): ), ) parser.add_argument( - "--center_crop", action="store_true", help="Whether to center crop images before resizing to resolution" + "--center_crop", + default=False, + action="store_true", + help="Whether to center crop images before resizing to resolution. If not set, images are randomly cropped.", ) parser.add_argument( "--train_text_encoder",