Modelpublic

atrost/climbmix-baseline-small-asymmetric-94m-1p2b-h100

sourceHugging Facecc-by-nc-4.0updated 5mo agoView on Hugging Face

0likes6downloads

Model Card

ClimbMix Baseline Small Asymmetric 94M

Exact-shape asymmetric nested baseline trained on 1.2B tokens from nvidia/Nemotron-ClimbMix.

Loading

python

from transformers import AutoModelForCausalLM, AutoTokenizer

repo_id = "atrost/climbmix-baseline-small-asymmetric-94m-1p2b-h100"
tokenizer = AutoTokenizer.from_pretrained(repo_id)
model = AutoModelForCausalLM.from_pretrained(
    repo_id,
    trust_remote_code=True,
    torch_dtype="auto",
)

trust_remote_code=True is required for the custom StairFormer/asymmetric checkpoints and harmless for the dense Llama baseline.

Source revision

This clean repo was split out from atrost/climbmix-llama-288m-2p8b-h100 at commit eb0352c.