fix: README_flux2.md fsdp config error - #14806
Thundermean-sky wants to merge 1 commit into
Conversation
|
Note: |
|
Hi @Thundermean-sky, thanks for the PR! It does not appear to link an issue it fixes. If this PR addresses an existing issue, please add a closing keyword (e.g. Please note that PRs without a linked issue are likely to be automatically closed 10 days after this notice. Once the PR links an issue (or gets the |
|
Hi, thanks for the reminder! This PR is a small documentation fix for incorrect FSDP config examples in the README (specifically Could a maintainer please add the
|
What does this PR do?
Fix incorrect FSDP configuration examples in the examples/dreambooth/README_flux2.md.
While training, I encountered two issues with the config examples in the README:
fsdp_auto_wrap_policywas incorrect, which caused training to fail.fsdp_transformer_layer_cls_to_wrap(Flux2TransformerBlock, Flux2SingleTransformerBlock). Removing the space resolved the error.After correcting both parameters, the training runs successfully.
This PR updates the README with the corrected FSDP configuration examples.
Before submitting
self-reviewskill on the diff?documentation guidelines, and
here are tips on formatting docstrings.
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.