easydel.modules.clip.modeling_clip_flax

easydel.modules.clip.modeling_clip_flax#

class easydel.modules.clip.modeling_clip_flax.CLIPAttention(*args: Any, **kwargs: Any)[source]#

Bases: FlaxAttentionModule

class easydel.modules.clip.modeling_clip_flax.CLIPEncoder(*args: Any, **kwargs: Any)[source]#

Bases: Module

property causal_mask#
class easydel.modules.clip.modeling_clip_flax.CLIPEncoderLayer(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.clip.modeling_clip_flax.CLIPForImageClassification(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.clip.modeling_clip_flax.CLIPMLP(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.clip.modeling_clip_flax.CLIPModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

compute_loss(*, labels=None, loss_config=None, loss_kwargs=None, **batch) → Tuple[Any, FlaxCLIPOutput][source]#

basic compute_loss call

get_image_features(pixel_values: Union[Array, ndarray, bool, number])[source]#
get_text_features(input_ids: Union[Array, ndarray, bool, number], attention_mask: Optional[Union[Array, ndarray, bool, number]] = None, position_ids: Optional[Union[Array, ndarray, bool, number]] = None)[source]#
class easydel.modules.clip.modeling_clip_flax.CLIPTextEmbeddings(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.clip.modeling_clip_flax.CLIPTextModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.clip.modeling_clip_flax.CLIPTextModelWithProjection(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.clip.modeling_clip_flax.CLIPTextTransformer(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.clip.modeling_clip_flax.CLIPVisionEmbeddings(*args: Any, **kwargs: Any)[source]#

Bases: Module

class easydel.modules.clip.modeling_clip_flax.CLIPVisionModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

class easydel.modules.clip.modeling_clip_flax.CLIPVisionTransformer(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

easydel.modules.clip.modeling_clip_flax.clip_loss(similarity: Array) → Array[source]#
easydel.modules.clip.modeling_clip_flax.contrastive_loss(logits: Array) → Array[source]#
Contents