easydel.modules.siglip.modeling_siglip_flax

Contents

easydel.modules.siglip.modeling_siglip_flax#

class easydel.modules.siglip.modeling_siglip_flax.MultiheadAttention(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.siglip.modeling_siglip_flax.SiglipAttention(*args: Any, **kwargs: Any)[source]#: Bases: AttentionModule

class easydel.modules.siglip.modeling_siglip_flax.SiglipEncoder(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.siglip.modeling_siglip_flax.SiglipEncoderLayer(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.siglip.modeling_siglip_flax.SiglipForImageClassification(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.siglip.modeling_siglip_flax.SiglipMLP(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.siglip.modeling_siglip_flax.SiglipModel(*args: Any, **kwargs: Any)[source]#

Bases: EasyDeLBaseModule

get_image_features(pixel_values: Optional[Union[Array, ndarray, bool, number]] = None, output_attentions: Optional[bool] = None, output_hidden_states: Optional[bool] = None, interpolate_pos_encoding: bool = False) → Union[Array, ndarray, bool, number][source]#

get_text_features(input_ids: Optional[Union[Array, ndarray, bool, number]] = None, attention_mask: Optional[Union[Array, ndarray, bool, number]] = None, position_ids: Optional[Union[Array, ndarray, bool, number]] = None, output_attentions: Optional[bool] = None, output_hidden_states: Optional[bool] = None) → Union[Array, ndarray, bool, number][source]#

class easydel.modules.siglip.modeling_siglip_flax.SiglipMultiheadAttentionPoolingHead(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.siglip.modeling_siglip_flax.SiglipOutput(loss: Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number, NoneType] = None, logits_per_image: Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number] = None, logits_per_text: Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number] = None, text_embeds: Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number] = None, image_embeds: Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number] = None, text_model_output: easydel.infra.modeling_outputs.BaseModelOutputWithPooling = None, vision_model_output: easydel.infra.modeling_outputs.BaseModelOutputWithPooling = None)[source]#

Bases: ModelOutput

classmethod from_dict(data: Dict[str, Any]) → T#: Deserializes a dictionary into a PyTree object.

classmethod from_json(json_str: str) → T#: Deserializes a JSON string into a PyTree object.

image_embeds: Union[Array, ndarray, bool, number] = None#

logits_per_image: Union[Array, ndarray, bool, number] = None#

logits_per_text: Union[Array, ndarray, bool, number] = None#

loss: Optional[Union[Array, ndarray, bool, number]] = None#

replace(**kwargs)#: Creates a new instance with specified fields replaced.

text_embeds: Union[Array, ndarray, bool, number] = None#

text_model_output: BaseModelOutputWithPooling = None#

to_dict() → Dict[str, Any]#: Serializes the PyTree object to a dictionary.

to_json(**kwargs) → str#: Serializes the PyTree object to a JSON string.

to_tuple() → Tuple[Any][source]#: Convert self to a tuple containing all the attributes/keys that are not None.

vision_model_output: BaseModelOutputWithPooling = None#

class easydel.modules.siglip.modeling_siglip_flax.SiglipTextEmbeddings(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.siglip.modeling_siglip_flax.SiglipTextModel(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput(text_embeds: Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number, NoneType] = None, last_hidden_state: Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number] = None, hidden_states: Optional[Tuple[Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number], ...]] = None, attentions: Optional[Tuple[Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number], ...]] = None)[source]#

Bases: ModelOutput

attentions: Optional[Tuple[Union[Array, ndarray, bool, number], ...]] = None#

classmethod from_dict(data: Dict[str, Any]) → T#: Deserializes a dictionary into a PyTree object.

classmethod from_json(json_str: str) → T#: Deserializes a JSON string into a PyTree object.

hidden_states: Optional[Tuple[Union[Array, ndarray, bool, number], ...]] = None#

last_hidden_state: Union[Array, ndarray, bool, number] = None#

replace(**kwargs)#: Creates a new instance with specified fields replaced.

text_embeds: Optional[Union[Array, ndarray, bool, number]] = None#

to_dict() → Dict[str, Any]#: Serializes the PyTree object to a dictionary.

to_json(**kwargs) → str#: Serializes the PyTree object to a JSON string.

class easydel.modules.siglip.modeling_siglip_flax.SiglipTextTransformer(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.siglip.modeling_siglip_flax.SiglipVisionEmbeddings(*args: Any, **kwargs: Any)[source]#

Bases: Module

interpolate(embeddings: Union[Array, ndarray, bool, number], height: int, width: int)[source]#

class easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModel(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput(image_embeds: Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number, NoneType] = None, last_hidden_state: Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number] = None, hidden_states: Optional[Tuple[Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number], ...]] = None, attentions: Optional[Tuple[Union[jax.Array, numpy.ndarray, numpy.bool, numpy.number], ...]] = None)[source]#

Bases: ModelOutput

attentions: Optional[Tuple[Union[Array, ndarray, bool, number], ...]] = None#

classmethod from_dict(data: Dict[str, Any]) → T#: Deserializes a dictionary into a PyTree object.

classmethod from_json(json_str: str) → T#: Deserializes a JSON string into a PyTree object.

hidden_states: Optional[Tuple[Union[Array, ndarray, bool, number], ...]] = None#

image_embeds: Optional[Union[Array, ndarray, bool, number]] = None#

last_hidden_state: Union[Array, ndarray, bool, number] = None#

replace(**kwargs)#: Creates a new instance with specified fields replaced.

to_dict() → Dict[str, Any]#: Serializes the PyTree object to a dictionary.

to_json(**kwargs) → str#: Serializes the PyTree object to a JSON string.

class easydel.modules.siglip.modeling_siglip_flax.SiglipVisionTransformer(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule