bottom-3B / modeling_olmo.py
cbrownpinilla's picture
HF integration
5fc231f
raw
history blame contribute delete
131 Bytes
from hf_olmo import OLMoForCausalLM, OLMoConfig
from transformers import AutoModel
AutoModel.register(OLMoConfig, OLMoForCausalLM)