[Ernie 4.5] Post merge adaptations (#39664)
* ernie 4.5 fixes * Apply style fixes * fix --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
@@ -33,7 +33,7 @@ CONFIG_MAPPING = transformers.models.auto.configuration_auto.CONFIG_MAPPING
|
||||
|
||||
SPECIAL_CASES_TO_ALLOW = {
|
||||
"Ernie4_5Config": ["tie_word_embeddings"],
|
||||
"Ernie4_5_MoEConfig": ["tie_word_embeddings"],
|
||||
"Ernie4_5_MoeConfig": ["tie_word_embeddings"],
|
||||
"Lfm2Config": ["full_attn_idxs", "tie_word_embeddings"],
|
||||
# used internally during generation to provide the custom logit processors with their necessary information
|
||||
"DiaConfig": [
|
||||
|
||||
Reference in New Issue
Block a user