easydel.modules.cohere2.modeling_cohere2_flax

easydel.modules.cohere2.modeling_cohere2_flax#

class easydel.modules.cohere2.modeling_cohere2_flax.Cohere2Attention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.cohere2.modeling_cohere2_flax.Cohere2Block(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.cohere2.modeling_cohere2_flax.Cohere2ForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.cohere2.modeling_cohere2_flax.Cohere2ForSequenceClassification(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.cohere2.modeling_cohere2_flax.Cohere2LayerNorm(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.cohere2.modeling_cohere2_flax.Cohere2MLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.cohere2.modeling_cohere2_flax.Cohere2Model(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

Contents