From 03a8646dc776f048da0d24dfbd288c1222268cf9 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Sun, 28 Jan 2024 17:40:59 +0530 Subject: [PATCH] correct hflip arg --- .../research_projects/diffusion_dpo/train_diffusion_dpo_sdxl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.