[BUG]: AttributeError: 'GeminiDDP' object has no attribute 'module'

[dheerj188]

Is there an existing issue for this bug?

• [X] I have searched the existing issues

🐛 Describe the bug

I'm trying to run Colossal AI's example on bert in language repo and when I run it with gemini plugin on 2 GPUs I get this error.

Environment

No response

[wangbluo]

Hi, could you please offer more error stack or error screen? Also your script and environment.