Skip to content
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.

Sign up for GitHub

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

Jump to bottom

Add version checks for the import of DeepSpeed moe utils #2705

Merged
merged 2 commits into from
Apr 24, 2024
Merged

Add version checks for the import of DeepSpeed moe utils #2705

merged 2 commits into from
Apr 24, 2024

Conversation

pacman100
Copy link
Contributor

What does this PR do?

  1. Add version checks for the import of DeepSpeed moe utils.

muellerzr
muellerzr approved these changes Apr 24, 2024
View reviewed changes
Copy link
Collaborator

@muellerzr muellerzr left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! LG2M

src/accelerate/utils/dataclasses.py Outdated Show resolved Hide resolved
tests/deepspeed/test_deepspeed.py Outdated Show resolved Hide resolved
@pacman100 @muellerzr
Apply suggestions from code review
2d2722f 
Co-authored-by: Zach Mueller <muellerzr@gmail.com>
@pacman100 pacman100 merged commit 092c3af into huggingface:main Apr 24, 2024
23 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@muellerzr muellerzr muellerzr approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@pacman100 @muellerzr