from transformers import AutoConfig
from hf_olmo import OLMoConfig

AutoConfig.register("hf_olmo", OLMoConfig)