pyvene.models.llama.modelings_intervenable_llama.create_llama

Contents

pyvene.models.llama.modelings_intervenable_llama.create_llama#

create_llama(name='sharpbai/alpaca-7b-merged', cache_dir=None, dtype=torch.bfloat16, config=None)[source]#

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