Modelpublic

Codeseys/composer-replication-framework

sourceHugging Facemitupdated 4mo agoView on Hugging Face

0likes

rootcomposer_replicationtrainer__init__.py

__init__.py21 linesDownload Raw Back to trainer

1"""composer_replication.trainer — TRL GRPOTrainer subclass + data collator.2 3Per docs/INTEGRATION_ARCHITECTURE.md § "Recipe A".4Per docs/adrs/ADR-003 (also wraps DiLoCo when training distributed).5"""6from __future__ import annotations7 8from composer_replication.trainer.composer_trainer import (9    PO_OBJECTIVES,10    ComposerReplicationTrainer,11    make_dr_grpo_config,12    make_po_config,13)14 15__all__ = [16    "ComposerReplicationTrainer",17    "make_dr_grpo_config",18    "make_po_config",19    "PO_OBJECTIVES",20]21