transformers
ff7c92d7 - Add TensorParallelTesterMixin to OLMoE tests

Commit
5 days ago
Add TensorParallelTesterMixin to OLMoE tests Enable TP test coverage (test_tp_forward, test_tp_backward, test_tp_generation, test_tp_plan_matches_params) for OLMoE now that base_model_tp_plan is defined. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Author
Parents
Loading