easydel.modules.cohere.modeling_cohere_flax

class easydel.modules.cohere.modeling_cohere_flax.CohereAttention(*args: Any, **kwargs: Any)

Bases: FlaxAttentionModule

class easydel.modules.cohere.modeling_cohere_flax.CohereBlock(*args: Any, **kwargs: Any)

Bases: Module

class easydel.modules.cohere.modeling_cohere_flax.CohereForCausalLM(*args: Any, **kwargs: Any)

Bases: EasyDeLBaseModule

class easydel.modules.cohere.modeling_cohere_flax.CohereForSequenceClassification(*args: Any, **kwargs: Any)

Bases: EasyDeLBaseModule

class easydel.modules.cohere.modeling_cohere_flax.CohereMLP(*args: Any, **kwargs: Any)

Bases: Module

class easydel.modules.cohere.modeling_cohere_flax.CohereModel(*args: Any, **kwargs: Any)

Bases: EasyDeLBaseModule

class easydel.modules.cohere.modeling_cohere_flax.RMSNorm(*args: Any, **kwargs: Any)

Bases: Module

easydel.modules.cohere.modeling_cohere_flax.repeat_kv(x: Union[Array, ndarray, bool, number], n_rep: int) → Union[Array, ndarray, bool, number]