cerebras.modelzoo.tools.checkpoint_converters.bloom_hf_cs#

Classes

ConfigConverter_BloomLMHeadModel_CS19_CS20

Bloom uses the GPT2 backbone.

ConfigConverter_BloomModel_HF_CS17

ConfigConverter_BloomModel_HF_CS19

ConfigConverter_BloomModel_HF_CS20

ConfigConverter_BloomModel_HF_CS21

CS 2.1 config is the same as CS 2.0.

Converter_BloomLMHeadModel_CS19_CS20

Bloom uses the GPT2 backbone.

Converter_BloomLMHeadModel_CS20_CS21

Converter_BloomLMHeadModel_HF_CS17

Converter_BloomLMHeadModel_HF_CS19

Converter_BloomLMHeadModel_HF_CS20

Converter_BloomLMHeadModel_WithoutOptionalModel_HF_CS21

Converter_BloomModel_HF_CS17

Converter_BloomModel_HF_CS19

Converter_BloomModel_HF_CS20

Converter_BloomModel_WithoutOptionalModel_HF_CS21

Converter_Bloom_Attention_HF_CS17