[BUG] 'GatheredParameters' object is not callable

[LinB203]

Describe the bug

'GatheredParameters' object is not callable when we use zero3 in text-to-image.py

The context_manager has been initilized, why we need with context_manager(). I think it should be with context_manager.
After replacing with context_manager() with with context_manager, the problem is solved.

af2a237#diff-5c07adf115dbb937f26dbfedff2aef0d31b4e8e3a58ba9cfb4a0c9f64672a8c5R213

Reproduction

when we use zero3 in text-to-image.py

Logs

No response

System Info

latest diffusers version

Who can help?

@yiyixuxu @DN6 @sayakpaul

[sayakpaul]

Thanks for investigating this! Do you want to submit a PR?

[charchit7]

Hey @LinB203, thanks for identifying the issue. I've raised a PR on your behalf - I hope that's alright with you.