locuslab/base-smollm2-1.7b-score0_baseline20p_then_mix_rephrase123_with_refusal_mild45-600B-step-60000

SmolLM2 score0_baseline20p_then_mix_rephrase123_with_refusal_mild45-600B-step-60000 (Version: main)

Model Details

• Architecture: SmolLM2
• Parameters: 1.7B

Training Configuration

optimizer:
  class_path: torch.optim.AdamW
  init_args:
    lr: 0.0005
    weight_decay: 0.01
precision: bf16-mixed
seed: 42
train:
  global_batch_size: 1024
  max_seq_length: 2048
  max_tokens: 600000000000
  micro_batch_size: 8

Model Loading and Revision System

This repository hosts multiple revisions of the model. To load a specific revision, use the revision parameter. For example:

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("locuslab/score0_baseline20p_then_mix_rephrase123_with_refusal_mild45-600B-step-60000", revision="final")
tokenizer = AutoTokenizer.from_pretrained("locuslab/score0_baseline20p_then_mix_rephrase123_with_refusal_mild45-600B-step-60000", revision="final")

Replace "final" with the desired revision.