cerebras.modelzoo.tools.checkpoint_converters.gpt_neox_hf_cs#

Classes

ConfigConverter_GPT_Neox_HF_CS17

ConfigConverter_GPT_Neox_HF_CS18

ConfigConverter_GPT_Neox_HF_CS20

ConfigConverter_GPT_Neox_HF_CS21

ConfigConverter_GPT_Neox_Headless_CS18_CS20

NeoX uses the GPTJ backbone.

Converter_GPT_Neox_Attention_HF_CS17

Converter_GPT_Neox_Headless_HF_CS17

Converter_GPT_Neox_Headless_HF_CS18

Converter_GPT_Neox_Headless_HF_CS20

Converter_GPT_Neox_Headless_HF_CS21

Converter_GPT_Neox_LMHeadModel_CS18_CS20

NeoX uses the GPTJ backbone.

Converter_GPT_Neox_LMHeadModel_CS20_CS21

Converter_GPT_Neox_LMHeadModel_HF_CS17

Converter_GPT_Neox_LMHeadModel_HF_CS18

Converter_GPT_Neox_LMHeadModel_HF_CS20

Converter_GPT_Neox_LMHeadModel_HF_CS21