integrations.hatchery.trainer

integrations.hatchery.trainer

Remote trainer that dispatches to Tinker or Hatchery API.

Classes

Name	Description
HatcheryTrainer	Trainer that sends preprocessed batches to a remote training API.

HatcheryTrainer

integrations.hatchery.trainer.HatcheryTrainer(*args, **kwargs)

Trainer that sends preprocessed batches to a remote training API.

Replaces local forward/backward with remote API calls to Tinker or Hatchery. Uses axolotl’s full data preprocessing pipeline (tokenization, chat templates, packing, etc.) but offloads compute to remote GPUs.

Methods

Name	Description
save_model	Delegate to remote checkpoint save so HF callbacks create checkpoints.
train	Main training loop — sends batches to remote API.

save_model

integrations.hatchery.trainer.HatcheryTrainer.save_model(
    output_dir=None,
    _internal_call=False,
)

Delegate to remote checkpoint save so HF callbacks create checkpoints.

train

integrations.hatchery.trainer.HatcheryTrainer.train(
    resume_from_checkpoint=None,
    trial=None,
    ignore_keys_for_eval=None,
    **kwargs,
)

Main training loop — sends batches to remote API.