pyvene.models.gpt2.modelings_intervenable_gpt2.create_gpt2_lm

Contents

pyvene.models.gpt2.modelings_intervenable_gpt2.create_gpt2_lm#

create_gpt2_lm(name='gpt2', config=None, cache_dir=None)[source]#

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