mindspore-lab / mindone

one for all, Optimal generator with No Exception
Apache License 2.0
329 stars 63 forks source link

Fixes(sdxl): fix some bugs in mindone.diffusers sdxl training #447

Closed townwish4git closed 2 months ago

townwish4git commented 2 months ago

What does this PR do?

Fixes # (issue)

Fixed bugs in mindone.diffusers sdxl training to support:

Masked option 'bf16' for amp as MindSpore has issues with dtype inference in that case

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag members/contributors who may be interested in your PR.

@xxx