easydel.modules.xerxes2.modeling_xerxes2_flax

easydel.modules.xerxes2.modeling_xerxes2_flax#

class easydel.modules.xerxes2.modeling_xerxes2_flax.Xerxes2Attention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.xerxes2.modeling_xerxes2_flax.Xerxes2DecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.xerxes2.modeling_xerxes2_flax.Xerxes2ForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

init_cache(batch_size: int, max_length: int)[source]#
class easydel.modules.xerxes2.modeling_xerxes2_flax.Xerxes2MLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.xerxes2.modeling_xerxes2_flax.Xerxes2Model(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

property frequencies: Array#

Returns frequency values from the config.

Contents