correct hflip arg (#6743)

acd19627 · Sayak Paul · GitHub · 5b1b80a5 · acd19627
Unverified Commit acd19627 authored Jan 28, 2024 by Sayak Paul Committed by GitHub Jan 28, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/research_projects/diffusion_dpo/train_diffusion_dpo_sdxl.py ...search_projects/diffusion_dpo/train_diffusion_dpo_sdxl.py +1 -1

No files found.
--- 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.