Add a converter from mamba_ssm -> huggingface mamba #29705
amyeroberts
merged
18 commits into
huggingface:main
from
byi8220:mamba-converter
implement convert_mamba_ssm_checkpoint_to_pytorch
a056af19
Add test test_model_from_mamba_ssm_conversion
f1d9488b
moved convert_ssm_config_to_hf_config to inside mamba_ssm_available c…
1d8d1da0
fix skipif clause
4e1b9dde
moved skips to inside test since skipif decorator isn't working for s…
b280bf98
amyeroberts
commented
on 2024-03-19
Merge branch 'huggingface:main' into mamba-converter
d2298bfe
Added validation
765ea4dd
removed test
c85c4260
Merge branch 'mamba-converter' of https://github.com/byi8220/transfor…
df04dd6e
fixup
6c5a5489
only compare logits
c3fc0716
Merge branch 'huggingface:main' into mamba-converter
ccc85ace
Merge branch 'huggingface:main' into mamba-converter
c76082e4
remove weight rename
ed4bea8d
Merge branch 'huggingface:main' into mamba-converter
76ab0e8e
Update src/transformers/models/mamba/convert_mamba_ssm_checkpoint_to_…
91d5b162
nits
500a0fb5
Merge branch 'huggingface:main' into mamba-converter
e56c4145
Login to write a write a comment.
Login via GitHub