pyvene.models.gpt2.modelings_intervenable_gpt2.create_gpt2

Contents

pyvene.models.gpt2.modelings_intervenable_gpt2.create_gpt2#

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

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