easydel.modules.qwen2.modeling_qwen_flax

class easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention(*args: Any, **kwargs: Any)

Bases: FlaxAttentionModule

class easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer(*args: Any, **kwargs: Any)

Bases: Module

class easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForCausalLM(*args: Any, **kwargs: Any)

Bases: EasyDeLBaseModule

class easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForSequenceClassification(*args: Any, **kwargs: Any)

Bases: EasyDeLBaseModule

class easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP(*args: Any, **kwargs: Any)

Bases: Module

class easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model(*args: Any, **kwargs: Any)

Bases: EasyDeLBaseModule