[Model] FalconMamba Support (#9325)

This commit is contained in:
Dhia Eddine Rhaiem authored and GitHub committed 2024-10-21 12:50:16 -04:00
1 parent 496e991da8
commit f6b97293aa
5 files changed
+35 -12

No files matched your search

+5
View File
@@ -87,6 +87,11 @@ Text Generation
- :code:`tiiuae/falcon-7b`, :code:`tiiuae/falcon-40b`, :code:`tiiuae/falcon-rw-7b`, etc.
-
- ✅︎
* - :code:`FalconMambaForCausalLM`
- FalconMamba
- :code:`tiiuae/falcon-mamba-7b`, :code:`tiiuae/falcon-mamba-7b-instruct`, etc.
- ✅︎
-
* - :code:`GemmaForCausalLM`
- Gemma
- :code:`google/gemma-2b`, :code:`google/gemma-7b`, etc.