diff --git a/examples/research_projects/diffusion_dpo/train_diffusion_dpo_sdxl.py b/examples/research_projects/diffusion_dpo/train_diffusion_dpo_sdxl.py index 3fd3fc11ad19..f175ff94a864 100644 --- a/examples/research_projects/diffusion_dpo/train_diffusion_dpo_sdxl.py +++ b/examples/research_projects/diffusion_dpo/train_diffusion_dpo_sdxl.py @@ -741,7 +741,7 @@ def preprocess_train(examples): combined_im = train_resize(combined_im) # Flipping. - if not args.no_flip and random.random() < 0.5: + if not args.no_hflip and random.random() < 0.5: combined_im = train_flip(combined_im) # Cropping.