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

Corrected the model checkpoint.
huggingface · Aug 15, 2024 · ab7e893 · ab7e893
1 parent e840127
commit ab7e893
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/falcon_mamba/modeling_falcon_mamba.py b/src/transformers/models/falcon_mamba/modeling_falcon_mamba.py
@@ -59,7 +59,7 @@
     (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"