repetition-experiments
/

test

test / modeling_olmo.py

Upload folder using huggingface_hub

d97faa2 verified over 1 year ago

131 Bytes

from hf_olmo import OLMoForCausalLM, OLMoConfig
from transformers import AutoModel

AutoModel.register(OLMoConfig, OLMoForCausalLM)

	from hf_olmo import OLMoForCausalLM, OLMoConfig
	from transformers import AutoModel

	AutoModel.register(OLMoConfig, OLMoForCausalLM)