easydel.modules.qwen2_moe.modeling_qwen2_moe_flax

easydel.modules.qwen2_moe.modeling_qwen2_moe_flax#

class easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForSequenceClassification(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeMLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeSparseMoeBlock(*args: Any, **kwargs: Any)[source]#

Bases: Module

Contents