from open_lm.utils.transformers.hf_model import OpenLMforCausalLM, OpenLMModel
from transformers import AutoModelForCausalLM

# AutoModelForCausalLM.register("openlm", OpenLMforCausalLM)