Joseph Enguehard 07bf2dff78 Add TokenClassification for Mistral, Mixtral and Qwen2 (#29878) ... * Add MistralForTokenClassification * Add tests and docs * Add token classification for Mixtral and Qwen2 * Save llma for token classification draft * Add token classification support for Llama, Gemma, Persimmon, StableLm and StarCoder2 * Formatting * Add token classification support for Qwen2Moe model * Add dropout layer to each ForTokenClassification model * Add copied from in tests * Update src/transformers/models/llama/modeling_llama.py Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> * Propagate suggested changes * Style --------- Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>		2024-05-20 10:06:57 +02:00
..
__init__.py	Starcoder2 model - bis (#29215)	2024-02-28 01:24:34 +01:00
test_modeling_starcoder2.py	Add TokenClassification for Mistral, Mixtral and Qwen2 (#29878)	2024-05-20 10:06:57 +02:00