easydel.modules.olmo2.modeling_olmo2_flax

easydel.modules.olmo2.modeling_olmo2_flax#

class easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForSequenceClassification(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.olmo2.modeling_olmo2_flax.Olmo2MLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

Contents