easydel.modules.exaone.modeling_exaone_flax

easydel.modules.exaone.modeling_exaone_flax#

class easydel.modules.exaone.modeling_exaone_flax.ExaoneAttention(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.exaone.modeling_exaone_flax.ExaoneAttentionInner(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.exaone.modeling_exaone_flax.ExaoneDecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.exaone.modeling_exaone_flax.ExaoneForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.exaone.modeling_exaone_flax.ExaoneForSequenceClassification(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.exaone.modeling_exaone_flax.ExaoneGatedMLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.exaone.modeling_exaone_flax.ExaoneModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

property frequencies#

Returns frequency values from the config.

Contents