File size: 186 Bytes
792a89e |
1 2 3 4 |
from open_lm.utils.transformers.hf_model import OpenLMforCausalLM, OpenLMModel
from transformers import AutoModelForCausalLM
# AutoModelForCausalLM.register("openlm", OpenLMforCausalLM) |