riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix: Corrected falcon-mamba-7b model checkpoint name (#32837)

Browse Source 
Corrected the model checkpoint.
This commit is contained in:
Sai-Suraj-27 2024-08-15 22:33:18 +05:30 committed by GitHub
parent e840127370
commit ab7e893d09
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/falcon_mamba/modeling_falcon_mamba.py
View File

@ -59,7 +59,7 @@ is_fast_path_available = all(
(selective_state_update, selective_scan_fn, causal_conv1d_fn, causal_conv1d_update, mamba_inner_fn)
)
_CHECKPOINT_FOR_DOC = "tiiuae/falcon_mamba-7b"
_CHECKPOINT_FOR_DOC = "tiiuae/falcon-mamba-7b"
_CONFIG_FOR_DOC = "FalconMambaConfig"