AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Diffusion DPO] apply fixes from #6547 (#6668)

Browse Source 
apply fixes from #6547
This commit is contained in:
Sayak Paul
2024-01-22 20:14:54 +05:30
committed by GitHub
parent d66d554dc2
commit da95a28ff6
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/research_projects/diffusion_dpo/train_diffusion_dpo_sdxl.py
View File

@@ -740,6 +740,10 @@ def main(args):
# Resize.
combined_im = train_resize(combined_im)
# Flipping.
if not args.no_flip and random.random() < 0.5:
combined_im = train_flip(combined_im)
# Cropping.
if not args.random_crop:
y1 = max(0, int(round((combined_im.shape[1] - args.resolution) / 2.0)))
@@ -749,11 +753,6 @@ def main(args):
y1, x1, h, w = train_crop.get_params(combined_im, (args.resolution, args.resolution))
combined_im = crop(combined_im, y1, x1, h, w)
# Flipping.
if random.random() < 0.5:
x1 = combined_im.shape[2] - x1
combined_im = train_flip(combined_im)
crop_top_left = (y1, x1)
crop_top_lefts.append(crop_top_left)
combined_im = normalize(combined_im)