From acd1962769715cebee1bb8e87075db4e78490070 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Sun, 28 Jan 2024 17:42:34 +0530 Subject: [PATCH] correct hflip arg (#6743) --- .../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 3fd3fc11ad..f175ff94a8 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 main(args): 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.