pyvene.models.gemma2.modelings_intervenable_gemma2.create_gemma2

Contents

pyvene.models.gemma2.modelings_intervenable_gemma2.create_gemma2#

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

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