Skip to content

autoTP for Qwen#4902

Merged
mrwyattii merged 7 commits intodeepspeedai:masterfrom
inkcherry:autotp_qwen
Jan 11, 2024
Merged

autoTP for Qwen#4902
mrwyattii merged 7 commits intodeepspeedai:masterfrom
inkcherry:autotp_qwen

Conversation

@inkcherry
Copy link
Contributor

Enabled autoTP for the Qwen model, added some module matching, and adjusted TP-related variables. Verification was conducted on Qwen-1_8B and Qwen-72B-chat.

@mrwyattii mrwyattii enabled auto-merge January 10, 2024 18:31
@inkcherry
Copy link
Contributor Author

It seems that the failure did not come from this PR, and I can pass it locally. Could you please check it again? Thanks! @mrwyattii

@mrwyattii mrwyattii added this pull request to the merge queue Jan 11, 2024
Merged via the queue into deepspeedai:master with commit ee7db48 Jan 11, 2024
mauryaavinash95 pushed a commit to mauryaavinash95/DeepSpeed that referenced this pull request Feb 17, 2024
Enabled autoTP for the Qwen model, added some module matching, and
adjusted TP-related variables. Verification was conducted on Qwen-1_8B
and Qwen-72B-chat.

---------

Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>
Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
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.

3 participants