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.

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

<fill-mask>Modify codes so that different accelerators can be called according to specific device conditions #863

Merged
merged 3 commits into from
Feb 26, 2024

Conversation

foin6
Copy link
Contributor

@foin6 foin6 commented Feb 26, 2024

Modifications in: DeepSpeedExamples/inference/huggingface/fill-mask :

  1. use "deepspeed.accelerator.get_accelerator()" to replace "cuda" specific codes;

p.s. Install intel-extension-for-deepspeed and intel-extension-for-pytorch before running.

@foin6
Copy link
Contributor Author

foin6 commented Feb 26, 2024

Hi @tjruwase, please take some time to review this PR, thanks!

@tjruwase tjruwase merged commit 6540db6 into microsoft:master Feb 26, 2024
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants