[Model] Support Pixtral models in the HF Transformers format (#9036)

2024-10-18 15:29:56 -04:00
parent 67a7e5ef38
commit 3921a2f29e
7 changed files with 503 additions and 12 deletions
--- a/docs/source/models/supported_models.rst
+++ b/docs/source/models/supported_models.rst
@@ -437,7 +437,7 @@ Text Generation
  * - :code:`PixtralForConditionalGeneration`
    - Pixtral
    - T + I\ :sup:`+`
-    - :code:`mistralai/Pixtral-12B-2409`
+    - :code:`mistralai/Pixtral-12B-2409`, :code:`mistral-community/pixtral-12b` etc.
    -
    - ✅︎
  * - :code:`QWenLMHeadModel`