-
Notifications
You must be signed in to change notification settings - Fork 3.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add support for colossalai 0.1.11 #15888
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Mind adding some unit tests ?
@tchaton It is a good suggestion. I think I can add some new unit tests. |
Head branch was pushed to by a user without write access
The mypy checks are failing: https://github.com/Lightning-AI/lightning/actions/runs/3711224217/jobs/6291426057 |
Co-authored-by: Carlos Mocholí <[email protected]>
…strategy/colossalai
seems there are no stable release yet so going to temporarily drop testing in #16122 |
Co-authored-by: Carlos Mocholí <[email protected]> Co-authored-by: Jirka Borovec <[email protected]>
Co-authored-by: Carlos Mocholí <[email protected]> Co-authored-by: Jirka Borovec <[email protected]>
Co-authored-by: Carlos Mocholí <[email protected]> Co-authored-by: Jirka Borovec <[email protected]>
Co-authored-by: Carlos Mocholí <[email protected]> Co-authored-by: Jirka Borovec <[email protected]>
Co-authored-by: Carlos Mocholí <[email protected]> Co-authored-by: Jirka Borovec <[email protected]>
Co-authored-by: Carlos Mocholí <[email protected]> Co-authored-by: Jirka Borovec <[email protected]>
What does this PR do?
Update the APIs in ColossalAIStrategy for its new version.
Since the API of Gemini in ColossalAI has updated recently, it becames incompatible with Lightning.
For compatibility with the upcoming version 0.1.11, I updated the ColossalAIStrategy.
Does your PR introduce any breaking changes? If yes, please list them.
Before submitting
PR review
Anyone in the community is welcome to review the PR.
Before you start reviewing, make sure you have read the review guidelines. In short, see the following bullet-list:
Did you have fun?
Make sure you had fun coding 🙃