ColossalAI
ColossalAI copied to clipboard
Gemini_auto error
🐛 Describe the bug
When I enable the optimization options inside the gemini_auto plugin, I encounter errors, such as TypeError: GeminiPlugin.init() got an unexpected keyword argument 'enable_flash_attention'.
Environment
No response
Hi, could you please share the traceback? I'll look into this issue.