easydel.modules.openelm.modeling_openelm_flax

easydel.modules.openelm.modeling_openelm_flax#

class easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.openelm.modeling_openelm_flax.OpenELMForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.openelm.modeling_openelm_flax.OpenELMModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

property frequencies#

Returns frequency values from the config.

class easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

Contents