Megatron-DeepSpeed
[BNB] integrate `StableEmbeding` into `VocabParallelEmbedding` logic
#182
Merged

[BNB] integrate `StableEmbeding` into `VocabParallelEmbedding` logic #182

stas00 merged 5 commits into main from bnb-stable-embed
stas00
stas00 move checks into args; undo StableEmbedding
2f034843
stas00 integrate StableEmedding into Embedding
dab20642
stas00
stas00 commented on 2021-11-07
TimDettmers
stas00 cleanup
1bba3748
stas00 ensure tp>1 is first for bnb
6306e1d8
stas00 Merge remote-tracking branch 'origin/main' into bnb-stable-embed
1dfaf541
stas00 stas00 merged a34ca7f2 into main 4 years ago
stas00 stas00 deleted the bnb-stable-embed branch 4 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone