easydel.modules.gemma2.modeling_gemma2_flax

easydel.modules.gemma2.modeling_gemma2_flax#

class easydel.modules.gemma2.modeling_gemma2_flax.Gemma2Attention(*args: Any, **kwargs: Any)[source]#

Bases: AttentionModule

class easydel.modules.gemma2.modeling_gemma2_flax.Gemma2DecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.gemma2.modeling_gemma2_flax.Gemma2ForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

Gemma2 model with a language modeling head for causal language modeling tasks.

This model extends the base Gemma2Model by incorporating a linear language modeling head on top of the base model, designed for generative tasks and text generation.

class easydel.modules.gemma2.modeling_gemma2_flax.Gemma2ForSequenceClassification(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.gemma2.modeling_gemma2_flax.Gemma2MLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.gemma2.modeling_gemma2_flax.Gemma2Model(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.gemma2.modeling_gemma2_flax.Gemma2RMSNorm(*args: Any, **kwargs: Any)[source]#

Bases: Module

Contents