Skip to content

Contribution: SLURM-based OLMo-3 SFT workflows + evaluation pipeline w/o Beaker infra dependency #1325

Description

@ferreirafabio

Hi,

I was in touch with @jacob-morrison a few weeks ago when I was trying to reproduce the OLMo-3 SFT checkpoints.

I've been working on finetuning OLMo-3 7B from base to instruct using OLMo-core + open-instruct without the Beaker infra dependency, and I've put together some documented workflows that might be useful for others in the community. I wanted to check if any of this would be a good fit for a PR. I was able to reproduce the performance of the 7b-instruct-sft checkpoint.

What I have:

1. End-to-end SLURM training workflows

2. Jupyter notebook workflows documenting the complete process:

3. Evaluation pipeline

4. Model testing utilities

  • Checkpoint conversion (OLMo-core to HuggingFace)
  • Test scripts for verifying trained models

Everything is documented with READMEs explaining the setup and usage (see MODEL_TESTING_GUIDE.md). The main value I see is having a reproducible, well-documented path for SLURM users to finetune OLMo-3 without needing Beaker/cloud infrastructure.

Would any of this be useful for open-instruct? Happy to clean things up and submit a PR if there's interest, or just keep it as a reference if it's not a fit.

Thanks!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions