easydel.modules.stablelm.modeling_stablelm_flax

easydel.modules.stablelm.modeling_stablelm_flax#

class easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.stablelm.modeling_stablelm_flax.StableLmForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.stablelm.modeling_stablelm_flax.StableLmLayerNormPerHead(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.stablelm.modeling_stablelm_flax.StableLmMLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

property frequencies#

Returns frequency values from the config.

Contents