[Model] Support Cohere2ForCausalLM (Cohere R7B) (#11203)

2024-12-16 11:56:19 +02:00
parent b3b1526f03
commit bddbbcb132
5 changed files with 26 additions and 4 deletions
--- a/docs/source/models/supported_models.rst
+++ b/docs/source/models/supported_models.rst
@@ -118,9 +118,9 @@ Text Generation (``--task generate``)
    - :code:`THUDM/chatglm2-6b`, :code:`THUDM/chatglm3-6b`, etc.
    - ✅︎
    - ✅︎
-  * - :code:`CohereForCausalLM`
+  * - :code:`CohereForCausalLM`,:code:`Cohere2ForCausalLM`
    - Command-R
-    - :code:`CohereForAI/c4ai-command-r-v01`, etc.
+    - :code:`CohereForAI/c4ai-command-r-v01`, :code:`CohereForAI/c4ai-command-r7b-12-2024`, etc.
    - ✅︎
    - ✅︎
  * - :code:`DbrxForCausalLM`