cerebras.modelzoo.tools.checkpoint_converters.gpt2_hf_cs#

Classes

ConfigConverter_GPT2Model_CS18_CS20

ConfigConverter_GPT2Model_CS20_CS21

ConfigConverter_GPT2Model_HF_CS17

ConfigConverter_GPT2Model_HF_CS18

ConfigConverter_GPT2Model_HF_CS20

ConfigConverter_GPT2Model_HF_CS21

CS 2.1 config is the same as CS 2.0

Converter_GPT2LMHeadModel_CS18_CS20

Converter_GPT2LMHeadModel_CS20_CS21

Converter_GPT2LMHeadModel_HF_CS17

Converter_GPT2LMHeadModel_HF_CS18

Converter_GPT2LMHeadModel_HF_CS20

Converter_GPT2LMHeadModel_WithoutOptionalModel_HF_CS21

Converter_GPT2Model_HF_CS17

Converter_GPT2Model_HF_CS20

Converter_GPT2Model_WithoutOptionalModel_HF_CS21

Converter_GPT2_Attention_HF_CS17