| | --- |
| | license: apache-2.0 |
| | --- |
| | # ZSL Checkpoints |
| |
|
| | ## Repository structure |
| | Each checkpoint is saved in a branch of this repository, named `{model}-{size}-step{step}`. |
| |
|
| | Each branch contains the following files: |
| | ``` |
| | config.yaml # training config |
| | model.pt # model weights |
| | optim.pt # optimizer weights |
| | train.pt # trainer params |
| | ``` |
| |
|
| | ## How to download checkpoints |
| | To download a file for a specific checkpoint use the branch name as the revision, e.g. |
| | ```python |
| | from huggingface_hub import hf_hub_download |
| | |
| | hf_hub_download(repo_id="mirandrom/zsl-checkpoints", filename="model.pt", revision="OLMo-14M-step1024") |
| | ``` |
| |
|
| | See [huggingface_hub](https://huggingface.co/docs/huggingface_hub/main/guides/download) for more information |
| |
|
