Add `base_model_tp_plan` to `OlmoeConfig` (#44668)
* Rebase: Add base_model_tp_plan to OlmoeConfig (dataclass style)
Rebased onto main after configs were migrated to dataclasses.
Adds base_model_tp_plan as a class attribute and TensorParallelTesterMixin
to the OLMoE test suite.
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
* Apply repo consistency fixes
* review: update src/transformers/models/olmoe/configuration_olmoe.py
Co-authored-by: Cyril Vallez <cyril.vallez@huggingface.co>
* review: use correct pattern for OlmoeModelTester class
---------
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Cyril Vallez <cyril.vallez@huggingface.co>
