easydel.modules.olmo.modeling_olmo_flax

easydel.modules.olmo.modeling_olmo_flax#

class easydel.modules.olmo.modeling_olmo_flax.FlaxOlmoDecoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.olmo.modeling_olmo_flax.OlmoAttention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.olmo.modeling_olmo_flax.OlmoForCausalLM(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.olmo.modeling_olmo_flax.OlmoForSequenceClassification(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.olmo.modeling_olmo_flax.OlmoMLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.olmo.modeling_olmo_flax.OlmoModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

Contents