sinapsis-huggingface 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: sinapsis-huggingface
3
- Version: 0.2.1
3
+ Version: 0.2.3
4
4
  Summary: Package for HuggingFace-based templates
5
5
  Author-email: SinapsisAI <dev@sinapsis.tech>
6
6
  Project-URL: Homepage, https://sinapsis.tech
@@ -1,38 +1,38 @@
1
- sinapsis_huggingface-0.2.1.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
1
+ sinapsis_huggingface-0.2.3.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
2
2
  sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
3
  sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/__init__.py,sha256=9FHbS4hse9WIE-1a5jJlG-23gB3wahlULANJAWQ464c,947
4
- sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/base_diffusers.py,sha256=KQJcntbnRjLIbV_C9hyRMHBAPVMb7PGG2DhA2a8_J_g,8295
4
+ sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/base_diffusers.py,sha256=Q70PxHKRYbz0LBS4hB3cW8pten72su4P2HCUa5X9xp8,8406
5
5
  sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/image_to_image_diffusers.py,sha256=pKPkHlxMAetVaSfUTHJ1Uu44OyYfO2d2FwWdhQX8Hhs,6312
6
6
  sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/image_to_video_gen_xl_diffusers.py,sha256=FqQuBBSSmI1wgKi_i1p5XYmyEIaq0JkXnMCu8VDMizQ,2112
7
7
  sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/inpainting_diffusers.py,sha256=06hHRKdKcFFEiY0sl0r3Wu58WlJjNmOIiW-CPNwb78g,13540
8
8
  sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/text_to_image_diffusers.py,sha256=3jIcM4SGqOtkcP7kH-D_3U0RaSf4beEjPrKsoGLAbSM,3091
9
9
  sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
10
  sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/templates/__init__.py,sha256=PJ7eLGjus8R7UBuXsI0u7cRbEZV5V6NWm5vzBj8453E,799
11
- sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/templates/hugging_face_embedding_extractor.py,sha256=HIJpUVfVKO-KXNBLllFWuubKmOW-8azRKoKW5KxFBWI,3308
11
+ sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/templates/hugging_face_embedding_extractor.py,sha256=nlRgz01whEPLZYjhI_R5BSBpTykPGgOsfnsQvW9NZDk,3417
12
12
  sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/templates/speaker_embedding_from_audio.py,sha256=4Zj78asL9oV-gb8n-P5fQtOwUixCJAyjNNxinV_7qWI,6160
13
- sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/templates/speaker_embedding_from_dataset.py,sha256=S-CJXihy03QC1bqRFzX5l8Gf5O6VJdD-x1V6rvs6o4U,3577
13
+ sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/templates/speaker_embedding_from_dataset.py,sha256=6Px0WUju4IlfFGtTxRVRZQahU-RGX95n_t4L8Q8AqSU,3679
14
14
  sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
15
15
  sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
16
16
  sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/helpers/grounding_dino_keys.py,sha256=XacOtRkd099sM64pXg_N0oxtnBCQHgY6N6XAcI4WCzo,791
17
17
  sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/__init__.py,sha256=PbNLyfrBTHZOmwF6mNFYLikFWEnVq9s-eFsRjFzIIWk,791
18
- sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/grounding_dino.py,sha256=gNQZLTUAyW8Vjig14eySE-GA19U9c53rdBYguKEcCvE,13329
18
+ sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/grounding_dino.py,sha256=Ei66fNa1iY6y3_2dblCQUz8Pv6Mws01KWVtHwHiWzFM,13448
19
19
  sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/grounding_dino_classification.py,sha256=_5onYorZaLKczULwqXHUMTAIJsduqaGqYzxaAcPbKak,8868
20
20
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
21
21
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/helpers/__init__.py,sha256=RYEd6xTaVlItleSPoq9RVJIFgXfY6aOHqy2SIO7zwjc,168
22
22
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/helpers/text_to_sentences.py,sha256=teaJXoTAVzGwar9gxenBabkA9VBJd-VAxsNXlzkKMuU,1676
23
23
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/__init__.py,sha256=3BgUm6C_tRgzxh2ADMBcu6OHzR-U5Tl1eFVtU0PwxB0,1095
24
- sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/base_transformers.py,sha256=GYijvmfGUBxn-Jdja-3bXj2pCaHW7Z0X0e4zJWcOLY0,5705
24
+ sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/base_transformers.py,sha256=PIeeQnL2vGQmEl4E5bnMEWqkN8ajiuG9wNyUg--lYvo,5749
25
25
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/image_to_text_transformers.py,sha256=XRZLw-Gk-r_aIQ1MJq2ZxUBxzUBet6YRMgFIZMrcYDk,2604
26
26
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/speech_to_text_transformers.py,sha256=3eBSyoUaq8N8mZbjRmo6zv3spfXCcxkurN9xNnsqWxU,2351
27
27
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/summarization_transformers.py,sha256=zTkt-Th7SjDe7AOXHtD710xKGBcXV-pcShzAthm2UPk,2016
28
28
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/text_to_speech_transformers.py,sha256=sjQlH0r-_eWVyVDNBZzEqXi2ZNhdRxJgBdphg2sChF4,5847
29
29
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/translation_transformers.py,sha256=-Q5KAa3sGSQWME8KHzY_MTOhxq5avwFIuwYPAdQ6COA,2548
30
- sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_base.py,sha256=izEncb4aBJMaJkDW7_3913XtO-OgCeJiRxCHBB0fycU,3754
30
+ sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_base.py,sha256=1R-nP3DGodYsb0LXVFrmx-K7eWo9an1nxbWscFL7mQg,3862
31
31
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_detection.py,sha256=Pur2GwPJshTzh6Fu7CAm2mgG4cutx_cZDMCtOCGvoIw,4134
32
32
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_inference.py,sha256=y-a90mQ4frr08YPSyslBXOf5vG5Ec0vAOSx5ZWPjy8U,10399
33
33
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/thirdparty/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
34
34
  sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/thirdparty/helpers.py,sha256=IGeYd5U2xpimpwTQW_5xm1pUYB5tqHlpq-fjwBHI4gY,2187
35
- sinapsis_huggingface-0.2.1.dist-info/METADATA,sha256=2BkSHYUivGoNxrrl73Vl9T4kv1Z_w7hJ_ndNIkXvjj8,11400
36
- sinapsis_huggingface-0.2.1.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
37
- sinapsis_huggingface-0.2.1.dist-info/top_level.txt,sha256=ZxHwnMjSWRceQL_6-B7GJBPxQWdlwkba-SYMVufhj5s,133
38
- sinapsis_huggingface-0.2.1.dist-info/RECORD,,
35
+ sinapsis_huggingface-0.2.3.dist-info/METADATA,sha256=5p5OIZ3DLlFahGx_ksJvxChuEV-P09hcuc7k2xUkENM,11400
36
+ sinapsis_huggingface-0.2.3.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
37
+ sinapsis_huggingface-0.2.3.dist-info/top_level.txt,sha256=ZxHwnMjSWRceQL_6-B7GJBPxQWdlwkba-SYMVufhj5s,133
38
+ sinapsis_huggingface-0.2.3.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.0.0)
2
+ Generator: setuptools (80.1.0)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5
 
@@ -11,8 +11,10 @@ from pydantic import BaseModel, ConfigDict
11
11
  from sinapsis_core.data_containers.data_packet import DataContainer, ImagePacket
12
12
  from sinapsis_core.template_base import Template
13
13
  from sinapsis_core.template_base.base_models import (
14
+ OutputTypes,
14
15
  TemplateAttributes,
15
16
  TemplateAttributeType,
17
+ UIPropertiesMetadata,
16
18
  )
17
19
  from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
18
20
 
@@ -64,7 +66,7 @@ class BaseDiffusers(Template, ABC):
64
66
  """
65
67
 
66
68
  AttributesBaseModel = BaseDiffusersAttributes
67
- CATEGORY = "Diffusers"
69
+ UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.IMAGE)
68
70
  TORCH_DTYPE: dict = TorchTypes().model_dump()
69
71
 
70
72
  def __init__(self, attributes: TemplateAttributeType) -> None:
@@ -6,8 +6,10 @@ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
6
6
  from sinapsis_core.data_containers.data_packet import DataContainer
7
7
  from sinapsis_core.template_base import Template
8
8
  from sinapsis_core.template_base.base_models import (
9
+ OutputTypes,
9
10
  TemplateAttributes,
10
11
  TemplateAttributeType,
12
+ UIPropertiesMetadata,
11
13
  )
12
14
 
13
15
 
@@ -44,7 +46,7 @@ class HuggingFaceEmbeddingExtractor(Template):
44
46
  """
45
47
 
46
48
  AttributesBaseModel = HuggingFaceEmbeddingExtractorAttributes
47
- CATEGORY = "Embeddings"
49
+ UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.TEXT)
48
50
 
49
51
  def __init__(self, attributes: TemplateAttributeType) -> None:
50
52
  super().__init__(attributes)
@@ -5,7 +5,7 @@ from typing import Literal
5
5
  from datasets import load_dataset
6
6
  from sinapsis_core.data_containers.data_packet import DataContainer
7
7
  from sinapsis_core.template_base import Template
8
- from sinapsis_core.template_base.base_models import TemplateAttributes
8
+ from sinapsis_core.template_base.base_models import OutputTypes, TemplateAttributes, UIPropertiesMetadata
9
9
  from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
10
10
 
11
11
 
@@ -63,7 +63,7 @@ class SpeakerEmbeddingFromDataset(Template):
63
63
  """
64
64
 
65
65
  AttributesBaseModel = SpeakerEmbeddingFromDatasetAttributes
66
- CATEGORY = "Embeddings"
66
+ UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.AUDIO)
67
67
 
68
68
  def execute(self, container: DataContainer) -> DataContainer:
69
69
  """Retrieve and attach speaker embeddings to specified packets in a DataContainer.
@@ -7,7 +7,12 @@ import torch
7
7
  from sinapsis_core.data_containers.annotations import BoundingBox, ImageAnnotations
8
8
  from sinapsis_core.data_containers.data_packet import DataContainer, ImagePacket
9
9
  from sinapsis_core.template_base import Template
10
- from sinapsis_core.template_base.base_models import TemplateAttributes, TemplateAttributeType
10
+ from sinapsis_core.template_base.base_models import (
11
+ OutputTypes,
12
+ TemplateAttributes,
13
+ TemplateAttributeType,
14
+ UIPropertiesMetadata,
15
+ )
11
16
  from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
12
17
  from transformers import (
13
18
  AutoModelForZeroShotObjectDetection,
@@ -75,7 +80,7 @@ class GroundingDINO(Template):
75
80
 
76
81
  """
77
82
 
78
- CATEGORY = "Grounding DINO"
83
+ UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.IMAGE)
79
84
  KEYS = GroundingDINOKeys()
80
85
 
81
86
  class AttributesBaseModel(GroundingBaseAttributes):
@@ -8,10 +8,7 @@ import torch
8
8
  from pydantic import Field
9
9
  from sinapsis_core.data_containers.data_packet import DataContainer
10
10
  from sinapsis_core.template_base import Template
11
- from sinapsis_core.template_base.base_models import (
12
- TemplateAttributes,
13
- TemplateAttributeType,
14
- )
11
+ from sinapsis_core.template_base.base_models import TemplateAttributes, TemplateAttributeType, UIPropertiesMetadata
15
12
  from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
16
13
  from transformers import AutoProcessor, pipeline
17
14
  from transformers.pipelines import Pipeline
@@ -56,7 +53,7 @@ class TransformersBase(Template):
56
53
  """
57
54
 
58
55
  AttributesBaseModel = TransformersBaseAttributes
59
- CATEGORY = "Transformers"
56
+ UIProperties = UIPropertiesMetadata(category="Transformers")
60
57
 
61
58
  def __init__(self, attributes: TemplateAttributeType) -> None:
62
59
  super().__init__(attributes)
@@ -6,8 +6,10 @@ import torch
6
6
  from sinapsis_core.data_containers.data_packet import DataContainer
7
7
  from sinapsis_core.template_base import Template
8
8
  from sinapsis_core.template_base.base_models import (
9
+ OutputTypes,
9
10
  TemplateAttributes,
10
11
  TemplateAttributeType,
12
+ UIPropertiesMetadata,
11
13
  )
12
14
  from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
13
15
  from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
@@ -41,7 +43,7 @@ class PaliGemmaBase(Template):
41
43
  """Base class for PaliGemma implementations."""
42
44
 
43
45
  AttributesBaseModel = PaliGemmaBaseAttributes
44
- CATEGORY = "Transformers"
46
+ UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.IMAGE)
45
47
  _TORCH_DTYPE: ClassVar[dict[str, Any]] = {"float16": torch.float16, "float32": torch.float32}
46
48
 
47
49
  def __init__(self, attributes: TemplateAttributeType) -> None: