easydel.modules.phi.modeling_phi_flax

easydel.modules.phi.modeling_phi_flax#

class easydel.modules.phi.modeling_phi_flax.FlaxPhiDecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.phi.modeling_phi_flax.PhiAttention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.phi.modeling_phi_flax.PhiForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.phi.modeling_phi_flax.PhiMLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.phi.modeling_phi_flax.PhiModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

property frequencies#

Returns frequency values from the config.

Contents