Uploaded image for project: 'Red Hat Enterprise Linux AI'
  1. Red Hat Enterprise Linux AI
  2. RHELAI-3981

Create `Checkpointer` class

XMLWordPrintable

    • Sprint 2

      Goal:
      The Checkpointer class introduces a unified approach to our various checkpointing techniques. A user can pass in their checkpointing style (full_state or hf_format), and the checkpointer, via checkpointer.checkpoint, will save the model using the selected method and other techniques (LoRA).

      https://github.com/instructlab/training/pull/456

      Acceptance Criteria:
      Checkpointer class is introduced to the library which allows users to create an object that represents one of the checkpointing methodologies we support. a user should be able to use this object to run training and save checkpoints

              cdoern@redhat.com Charles Doern
              cdoern@redhat.com Charles Doern
              Votes:
              0 Vote for this issue
              Watchers:
              1 Start watching this issue

                Created:
                Updated: