easydel.modules.mistral.modeling_mistral_flax

easydel.modules.mistral.modeling_mistral_flax#

class easydel.modules.mistral.modeling_mistral_flax.MistralAttention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.mistral.modeling_mistral_flax.MistralDecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.mistral.modeling_mistral_flax.MistralForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.mistral.modeling_mistral_flax.MistralForSequenceClassification(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.mistral.modeling_mistral_flax.MistralMLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.mistral.modeling_mistral_flax.MistralModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

Contents