easydel.modules.gpt_j.modeling_gpt_j_flax

easydel.modules.gpt_j.modeling_gpt_j_flax#

class easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJAttention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJBlock(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJMLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

property frequencies#

Returns frequency values from the config.

Contents