easydel.modules.internlm2.modeling_internlm2_flax

easydel.modules.internlm2.modeling_internlm2_flax#

class easydel.modules.internlm2.modeling_internlm2_flax.FlaxInternLM2Block(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForSequenceClassification(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.internlm2.modeling_internlm2_flax.InternLM2MLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

Contents