Update train_dreambooth_colossalai.py

[haofanwang]

Fix a known bug in this example. From a recent update in colossalai.

[HuggingFaceDocBuilderDev]

The documentation is not available anymore as the PR was closed or merged.

[haofanwang]

@patrickvonplaten This update has been merged into ColossalAI, it should be fine to merge into diffusers.

[patrickvonplaten]

Hey @haofanwang,

Could you maybe run:

make style

once to fix the failing test?

[haofanwang]

Already done. But it seems that the testing is still fail? @patrickvonplaten

[isamu-isozaki]

@haofanwang I noticed that when you run the style check it modifies the other parts of the code too which causes the tests to fail interestingly enough. So I recommend doing git add on just the colossalai part and nothing else. Then I think it should be fixed.

[haofanwang]

Thanks. Will fix this in a new PR.