pyvene.models.olmo.modelings_intervenable_olmo.create_olmo

Contents

pyvene.models.olmo.modelings_intervenable_olmo.create_olmo#

create_olmo(name='allenai/OLMo-7B-0424-hf', cache_dir=None, dtype=torch.bfloat16, config=None, revision='main')[source]#

Creates a OLMo Causal LM model, config, and tokenizer from the given name and revision