Skip to content

[Device]Support DeepSpeedChat to run on different device besides cuda#736

Merged
tjruwase merged 2 commits into
deepspeedai:masterfrom
yisustc:sup_xpu
Sep 22, 2023
Merged

[Device]Support DeepSpeedChat to run on different device besides cuda#736
tjruwase merged 2 commits into
deepspeedai:masterfrom
yisustc:sup_xpu

Conversation

@yisustc

@yisustc yisustc commented Sep 20, 2023

Copy link
Copy Markdown
Contributor

This pr is to support DeepSpeedChat to run on different other device besides cuda, use get_accelerator from deepspeed to replace the fixed cuda.

Comment thread applications/DeepSpeed-Chat/training/step1_supervised_finetuning/prompt_eval.py Outdated
Comment thread applications/DeepSpeed-Chat/training/step2_reward_model_finetuning/rw_eval.py Outdated
@tjruwase

Copy link
Copy Markdown
Contributor

@ys950902, thanks! Please see the feedback.

@yisustc

yisustc commented Sep 22, 2023

Copy link
Copy Markdown
Contributor Author

Hi @tjruwase, many thanks with your feedback, I have updated the code with your suggestions, could you please take a look on it.

@tjruwase tjruwase merged commit db56381 into deepspeedai:master Sep 22, 2023
@yisustc yisustc deleted the sup_xpu branch September 22, 2023 16:42
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