mirandrom/zsl-checkpoints

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.

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 for more information