README.md

---
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