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: FlaxAttentionModule

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

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