Skip to content

Commit

Permalink
We use safetensors now.
Browse files Browse the repository at this point in the history
  • Loading branch information
dirkgr committed Nov 26, 2024
1 parent a6e6e2b commit 4f0d7d1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/Checkpoints.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ Unless otherwise specified, an OLMo checkpoint is assumed to be unsharded. OLMo
Each unsharded checkpoint directory consists of:

- `config.yaml`: the config at that training step.
- `model.pt`, `optim.pt`, `train.pt`: model, optimizer and training state at that training step.
- `model.pt`, `optim.pt`, `train.safetensors`: model, optimizer and training state at that training step.

URLs to checkpoints at intermediate steps of our official models' trainings can be found in the csv files under [`checkpoints/official/`](https://github.com/allenai/OLMo/blob/main/checkpoints/official). These 'directory' URLs cannot currently be directly accessed, but files within the directory are publicly accessible.

Expand Down

0 comments on commit 4f0d7d1

Please sign in to comment.