DeepSpeed
Add FALCON-40B Inference-Kernel Support
#3656
Open

Add FALCON-40B Inference-Kernel Support #3656

RezaYazdaniAminabadi wants to merge 15 commits into master from ds-inference/add-falcon-support
RezaYazdaniAminabadi
Add FALCON auto-tp support
8b0ddc8a
Add the FALCON-40B kernel support
216c1203
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from jeffra jeffra 2 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from mrwyattii mrwyattii 2 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from awan-10 awan-10 2 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from cmikeh2 cmikeh2 2 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from arashb arashb 2 years ago
RezaYazdaniAminabadi Merge branch 'master' into ds-inference/add-falcon-support
9b10c1a9
add missing flacon-ploicy file
cad6ebd4
Merge branch 'ds-inference/add-falcon-support' of github.com:microsof…
3016e1d6
remove the debug code
2d54af9a
add checkpoint loading for flacon model
0df4059d
Yard1
thies1006
Yard1
RezaYazdaniAminabadi
thies1006
Yard1
RezaYazdaniAminabadi
RezaYazdaniAminabadi
RezaYazdaniAminabadi Merge branch 'master' into ds-inference/add-falcon-support
468c68ec
thies1006
Yard1
RezaYazdaniAminabadi Merge branch 'master' into ds-inference/add-falcon-support
38b611de
lanking520
RezaYazdaniAminabadi
RezaYazdaniAminabadi Merge branch 'master' into ds-inference/add-falcon-support
f512587f
alexwong2024
ldong87
RezaYazdaniAminabadi
ldong87
add the load-checkpoint support for falcon-40b (reducing loading time…
8c5967f4
RezaYazdaniAminabadi
more fixes on the ckpt loading
fe4cd2ef
RezaYazdaniAminabadi
lanking520
ldong87
lanking520
RezaYazdaniAminabadi
RezaYazdaniAminabadi RezaYazdaniAminabadi closed this 2 years ago
dc3671
RezaYazdaniAminabadi
fix a few issues & add falcon-7b support with meta-tensor loading
2108c701
RezaYazdaniAminabadi RezaYazdaniAminabadi reopened this 2 years ago
add falcon test script
dfa4784c
RezaYazdaniAminabadi
some fixes for the the new Falcon Architecture
e1ba1077
dc3671
mynewstart

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone