Unverified Commit 0d4dfbbd authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Examples] fix: prior preservation setting in DreamBooth LoRA SDXL script. (#7242)



fix: prior preservation setting in DreamBooth LoRA SDXL script.
Co-authored-by: default avatarLinoy Tsaban <57615435+linoytsaban@users.noreply.github.com>
parent ada3bb94
...@@ -877,6 +877,8 @@ def collate_fn(examples, with_prior_preservation=False): ...@@ -877,6 +877,8 @@ def collate_fn(examples, with_prior_preservation=False):
if with_prior_preservation: if with_prior_preservation:
pixel_values += [example["class_images"] for example in examples] pixel_values += [example["class_images"] for example in examples]
prompts += [example["class_prompt"] for example in examples] prompts += [example["class_prompt"] for example in examples]
original_sizes += [example["original_size"] for example in examples]
crop_top_lefts += [example["crop_top_left"] for example in examples]
pixel_values = torch.stack(pixel_values) pixel_values = torch.stack(pixel_values)
pixel_values = pixel_values.to(memory_format=torch.contiguous_format).float() pixel_values = pixel_values.to(memory_format=torch.contiguous_format).float()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment