search

mindspore-lab / mindone

one for all, Optimal generator with No Exception
https://mindspore-lab.github.io/mindone/
Apache License 2.0
332 stars 63 forks source link

Scale lr & empty prompt #327

Closed townwish4git closed 5 months ago

townwish4git commented 5 months ago
  1. scale learning rate by global batch size, reference: https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_sdxl.py#L313
  2. set prompts empty randomly, reference: https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_sdxl.py#L219