easydel.modules.gpt2.modeling_gpt2_flax

easydel.modules.gpt2.modeling_gpt2_flax#

class easydel.modules.gpt2.modeling_gpt2_flax.Conv1D(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.gpt2.modeling_gpt2_flax.GPT2Attention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.gpt2.modeling_gpt2_flax.GPT2Block(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.gpt2.modeling_gpt2_flax.GPT2LMHeadModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.gpt2.modeling_gpt2_flax.GPT2MLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.gpt2.modeling_gpt2_flax.GPT2Model(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

Contents