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

SD-XL: Add support for long prompts training #331

Closed Cui-yshoho closed 5 months ago

Cui-yshoho commented 5 months ago
  1. add get_text_index, lpw_openclip_tokenize, pad_tokens, pre compile graph
  2. update embedding
  3. add the parameter --lpw True to use long prompts training
SamitHuang commented 5 months ago

Please supplement the readme doc on how-to enable long prompt training.