d97faa2
1
2
3
4
from hf_olmo import OLMoForCausalLM, OLMoConfig from transformers import AutoModel AutoModel.register(OLMoConfig, OLMoForCausalLM)
