pyvene.models.gpt_neo.modelings_intervenable_gpt_neo.create_gpt_neo

Contents

pyvene.models.gpt_neo.modelings_intervenable_gpt_neo.create_gpt_neo#

create_gpt_neo(name='roneneldan/TinyStories-33M', cache_dir=None)[source]#

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