
Michael WandandSigbjørn Skjæret
4988f6e866
Add arch support for cohere2-MoE (#24260)
* Add arch support for cohere2-MoE
* Removed redundant gating_func checks
* Changed ffn lookup to prefer prefix_dense_intermediate_size
* Renamed arch to cohere2moe
* Removed redundant lmhead check and chat template changes
* Removed lm_head.weight check from modify tensors, load output tensor not required, fallback to token_embd.weight
* Changed to (routed+shared)*0.5 for shared expert combined avg
* fixed sliding_window_pattern issue and pattern
* Fixed transformers crash 'first_k_dense_replace' error
* Remove comment
* Removed cohere2-moe as a tokenizer type and kept as tiny_aya. Renamed North-Mini-Code-1.0.
* Fixed MTP fail, changed to use iSWA
* Fixed remaining todos: cohere2moe renamed, changed swa parsing to use get_key_or_arr, removed extra get_arr use
* Force metadata usage
Co-authored-by: Sigbjørn Skjæret <[email protected]>
* Remove Cohere2 checkpoint comment
Co-authored-by: Sigbjørn Skjæret <[email protected]>
* Remove MTP comment
Co-authored-by: Sigbjørn Skjæret <[email protected]>
* Regenerate cohere2moe tokenizer hash
* Add cohere2moe to Llama Model Saver supported list
* Check for zerobios tensors and add support for Command to use LayerNorm
* Map expert_selection_fn to sigmoid in base.py instead of command.py
* use bools for foundnorm/foundnormrms
Co-authored-by: Sigbjørn Skjæret <[email protected]>
---------
Co-authored-by: Sigbjørn Skjæret <[email protected]>
2026-06-13 19:49:00 +02:00
..
2026-06-13 19:49:00 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-06-13 19:49:00 +02:00
2026-06-02 17:55:11 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-06-13 19:49:00 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-29 16:13:51 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-06-01 11:48:53 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-06-08 13:48:52 -07:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-06-05 17:44:59 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-21 00:35:37 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-06-12 10:21:06 +03:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-06-02 22:11:12 +03:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-06-07 21:41:39 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 18:38:39 +02:00
2026-05-25 14:16:11 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-06-02 17:44:35 +02:00
2026-05-15 15:18:12 +02:00
2026-05-26 07:57:38 +03:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00
2026-05-15 15:18:12 +02:00