easydel.modules.mixtral.modeling_mixtral_flax

Contents

easydel.modules.mixtral.modeling_mixtral_flax#

class easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention(*args: Any, **kwargs: Any)[source]#: Bases: FlaxAttentionModule

class easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.mixtral.modeling_mixtral_flax.MixtralSparseMoeBlock(*args: Any, **kwargs: Any)[source]#: Bases: Module