pyvene.models.gemma.modelings_intervenable_gemma.create_gemma

Contents

pyvene.models.gemma.modelings_intervenable_gemma.create_gemma#

create_gemma(name='google/gemma-2b-it', cache_dir=None, dtype=torch.bfloat16)[source]#

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