sinapsis-huggingface 0.2.11__py3-none-any.whl → 0.2.13__py3-none-any.whl

{sinapsis_huggingface-0.2.11.dist-info → sinapsis_huggingface-0.2.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-huggingface
-Version: 0.2.11
+Version: 0.2.13
 Summary: Package for HuggingFace-based templates
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech
@@ -23,6 +23,7 @@ Requires-Dist: sinapsis-huggingface-diffusers[all]; extra == "all"
 Requires-Dist: sinapsis-huggingface-embeddings; extra == "all"
 Requires-Dist: sinapsis-huggingface-grounding-dino; extra == "all"
 Requires-Dist: sinapsis-huggingface-transformers; extra == "all"
+Requires-Dist: sinapsis-huggingface-hub[all]; extra == "all"
 Requires-Dist: sinapsis-huggingface[gradio-app,sinapsis-data-writers]; extra == "all"
 Provides-Extra: sinapsis-data-writers
 Requires-Dist: sinapsis-data-writers>=0.1.0; extra == "sinapsis-data-writers"
@@ -60,6 +61,7 @@ This repo consists of different packages to handle huggingface tools for differe
 * <code>sinapsis-huggingface-embeddings</code>
 * <code>sinapsis-huggingface-grounding-dino</code>
 * <code>sinapsis-huggingface-transformers</code>
+* <code>sinapsis-huggingface-hub</code>
 
 Install using your package manager of choice. We encourage the use of <code>uv</code>
 
@@ -151,6 +153,17 @@ For more details and specific templates, see the [README.md](https://github.com/
 
 </details>
 
+<details>
+<summary id="uv"><strong><span style="font-size: 1.4em;">Sinapsis Hugging Face Hub</span></strong></summary>
+
+This sinapsis package offers templates to manage **datasets**, **models** and **spaces** with the Hugging Face Hub library. Currently it offers:
+
+- **HuggingFaceDownloader**: Downloads a repository snapshot from the Hugging Face Hub.
+
+For more details and specific templates, see the [README.md](https://github.com/Sinapsis-AI/sinapsis-huggingface/blob/main/packages/sinapsis_huggingface_hub/README.md).
+
+</details>
+
 For more details, see the [official documentation](https://docs.sinapsis.tech/docs)
 
 <h2 id="webapps">🌐 Webapps</h2>

{sinapsis_huggingface-0.2.11.dist-info → sinapsis_huggingface-0.2.13.dist-info}/RECORD

@@ -1,13 +1,13 @@
-sinapsis_huggingface-0.2.11.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+sinapsis_huggingface-0.2.13.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
 sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/helpers/tags.py,sha256=TFmVD7r70vKmpNqSweVGme4riZZiRQWIfxySTexyJp8,522
 sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/__init__.py,sha256=9FHbS4hse9WIE-1a5jJlG-23gB3wahlULANJAWQ464c,947
-sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/base_diffusers.py,sha256=OOb7xwHXnPifWWknrdYsk9i_U2dSdnBaeF6qDcaZ6xQ,10242
-sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/image_to_image_diffusers.py,sha256=OYT5fZBzCZoW7WTFi9kpvibGJw8wHTMm_O0eu74CRT4,6595
-sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/image_to_video_gen_xl_diffusers.py,sha256=tV5Ebv3-Qh6684WZQ7phE1B4QUGq0espICJFponhZ60,3590
-sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/inpainting_diffusers.py,sha256=aiGMXphGRjqBWDFnAanp-Iv-P_igmI6vj0Phg5JvZ3E,13800
-sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/text_to_image_diffusers.py,sha256=LUuI3A4MXcHZrq7mCPvkgmswRNicmIvhLiZ6bdNTpqM,3360
+sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/base_diffusers.py,sha256=a-ZvbRgAdOY89z5GCcBjq02TTAPhbH9KN8KGoMFB-LE,11530
+sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/image_to_image_diffusers.py,sha256=Dj-ytpfsTMbsoN81xSwYr5be7nsg3rqznUFEsnceJbY,6548
+sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/image_to_video_gen_xl_diffusers.py,sha256=4X_iWj4mCzYPBWbWP3Zy0UJsMw-PPReroqiQ2Ieodz0,3746
+sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/inpainting_diffusers.py,sha256=5r7l4F_0A6V6waCSaitJhrdeXfZsdq2kAosO3o1LmXI,13932
+sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/text_to_image_diffusers.py,sha256=IAGiFQJcXUqnhCTCfHCWjcYT9ar_1Y5Rn15ImCUiDks,3359
 sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/helpers/tags.py,sha256=zYMLpWXVATN67qFzJwclCQ9WE7TRTihPKKZ-vaH6nEU,284
@@ -19,25 +19,30 @@ sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/help
 sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/helpers/grounding_dino_keys.py,sha256=XacOtRkd099sM64pXg_N0oxtnBCQHgY6N6XAcI4WCzo,791
 sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/helpers/tags.py,sha256=q_QiumP-iCGLbiV0LOSc-Y4anUASmPwgplhnCXEiuXI,368
 sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/__init__.py,sha256=PbNLyfrBTHZOmwF6mNFYLikFWEnVq9s-eFsRjFzIIWk,791
-sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/grounding_dino.py,sha256=GvU80V7QTk3ZPeGzfcQLIjWVezU9ot98kQ5AXPgsM_E,14379
+sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/grounding_dino.py,sha256=Qucb8JcicjgovT1p32JuxVQt5k0h4dn3syegWE7Hcug,15154
 sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/grounding_dino_classification.py,sha256=UeOarFsC8z0jg5WtOO9NAvhYnOFpEaLkM75fsGhRiZI,9171
+sinapsis_huggingface_hub/src/sinapsis_huggingface_hub/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sinapsis_huggingface_hub/src/sinapsis_huggingface_hub/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sinapsis_huggingface_hub/src/sinapsis_huggingface_hub/helpers/tags.py,sha256=1SzO5cGKKGufgTIsN1_RYzsDPs99iowI7K3M1ZwtQSI,120
+sinapsis_huggingface_hub/src/sinapsis_huggingface_hub/templates/__init__.py,sha256=TsgxRtbtvFs2fwWLr5HuL3cLSll4Kbws-WOz1c7CZng,517
+sinapsis_huggingface_hub/src/sinapsis_huggingface_hub/templates/huggingface_downloader.py,sha256=mizdvhBmPFldBcZvqUNpvWe87-9IxMG7g8x7HsJzNnU,3442
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/helpers/__init__.py,sha256=RYEd6xTaVlItleSPoq9RVJIFgXfY6aOHqy2SIO7zwjc,168
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/helpers/tags.py,sha256=F6zVOBh8mVnl9AG0s_t1ftZ-1Fg1RoO3zgng9PxfLWk,652
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/helpers/text_to_sentences.py,sha256=teaJXoTAVzGwar9gxenBabkA9VBJd-VAxsNXlzkKMuU,1676
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/__init__.py,sha256=3BgUm6C_tRgzxh2ADMBcu6OHzR-U5Tl1eFVtU0PwxB0,1095
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/base_transformers.py,sha256=8vZBGQka2kSPYjbn5PbnT256_1UrJ099ygS6rQOMkmY,7647
-sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/image_to_text_transformers.py,sha256=6L4IUGv9l6Z3xQi5qZB70KPIa1BBUmckS1BhkmXbrjs,4272
-sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/speech_to_text_transformers.py,sha256=4lewS44TXjWoiPtcvNM9Xo1rvm8XkYTJtuduDXgbrXI,4062
-sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/summarization_transformers.py,sha256=XF7HYBNaEThyFJoqdWxb3mUqBtxEtnHN6mb6jwP_UEk,3732
-sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/text_to_speech_transformers.py,sha256=epvTmlxrRC1xeFiQzX0BJY9rHg6KtbYJSlKqbsLnE1k,6439
+sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/image_to_text_transformers.py,sha256=_t1oSfDNqpABGXoA1aRyZASWUoZzVYgtnpa8aEoQM6Y,4335
+sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/speech_to_text_transformers.py,sha256=y1FanicU5NpzIjxPor_8JNUBozGlmnQJkQW0FHAe6SE,4135
+sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/summarization_transformers.py,sha256=05YNUe1wx81dCrBagQ39ofueeG5bp0nZ7Mo9pjAVfPA,3894
+sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/text_to_speech_transformers.py,sha256=-NteA-dk1FPo1y93E5oHUXOTR0e5NrJIYlGBLGeFpqo,6534
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/translation_transformers.py,sha256=36ZjRzd8GRKq8s_b3_zkyK8BE7pmvyrtqi1_UNA0Fjg,4014
-sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_base.py,sha256=rcmnEo2o_xm7wExWfJDD9b8KQy0cgd2VeUyGNG6t4Aw,5037
+sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_base.py,sha256=18WkkpkfnPY39gpLRzgoISEkVxMgCXNOAhb9gumVO94,5104
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_detection.py,sha256=0bQB0rDU5foADo5IOe422UUJ1v6BZ3Q08esklwKOOQA,4210
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_inference.py,sha256=gC6RXqQi423uZr-GFpI_XzrRHjqMgJ7ry5MXC2-Y5N8,10894
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/thirdparty/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/thirdparty/helpers.py,sha256=IGeYd5U2xpimpwTQW_5xm1pUYB5tqHlpq-fjwBHI4gY,2187
-sinapsis_huggingface-0.2.11.dist-info/METADATA,sha256=MmqGcCSRmeJ1AricO6rQIrEjKMm8Q_P0KFGvJiDm2BI,11401
-sinapsis_huggingface-0.2.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sinapsis_huggingface-0.2.11.dist-info/top_level.txt,sha256=ZxHwnMjSWRceQL_6-B7GJBPxQWdlwkba-SYMVufhj5s,133
-sinapsis_huggingface-0.2.11.dist-info/RECORD,,
+sinapsis_huggingface-0.2.13.dist-info/METADATA,sha256=HFXHHJqNUCYFS7Ce7eyI4ToebhCOQfk9XTstBT_2nvI,12039
+sinapsis_huggingface-0.2.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sinapsis_huggingface-0.2.13.dist-info/top_level.txt,sha256=kb6h2cFbLt44Av6X418hZN7VYFbfBJCDpQGVETMvpvg,158
+sinapsis_huggingface-0.2.13.dist-info/RECORD,,

{sinapsis_huggingface-0.2.11.dist-info → sinapsis_huggingface-0.2.13.dist-info}/top_level.txt

@@ -1,4 +1,5 @@
 sinapsis_huggingface_diffusers
 sinapsis_huggingface_embeddings
 sinapsis_huggingface_grounding_dino
+sinapsis_huggingface_hub
 sinapsis_huggingface_transformers

sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/base_diffusers.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 
+import gc
 from abc import ABC, abstractmethod
 from typing import Literal
 
@@ -70,7 +71,7 @@ class BaseDiffusersAttributes(TemplateAttributes):
             Defaults to False.
     """
 
-    model_path: str
+    model_path: Literal["stable-diffusion-v1-5/stable-diffusion-v1-5", "stabilityai/stable-diffusion-2-1", "CompVis/stable-diffusion-v1-4", "Qwen/Qwen-Image-Edit"] = "stable-diffusion-v1-5/stable-diffusion-v1-5"
     model_cache_dir: str = str(SINAPSIS_CACHE_DIR)
     device: Literal["cuda", "cpu"]
     torch_dtype: Literal["float16", "float32"] = "float16"
@@ -107,7 +108,14 @@ class BaseDiffusers(Template, ABC):
 
     def __init__(self, attributes: TemplateAttributeType) -> None:
         super().__init__(attributes)
+        self.initialize()
 
+    def initialize(self) -> None:
+        """Initializes the template's common state for creation or reset.
+
+        This method is called by both `__init__` and `reset_state` to ensure
+        a consistent state. Can be overriden by subclasses for specific behaviour.
+        """
         self.pipeline = self._make_pipeline()
         self.pipeline.set_progress_bar_config(disable=True)
         self.num_images_per_prompt = self.attributes.generation_params.num_images_per_prompt
@@ -155,7 +163,7 @@ class BaseDiffusers(Template, ABC):
         pipeline_class = self._pipeline_class()
         return pipeline_class.from_pretrained(
             self.attributes.model_path,
-            torch_dtype=self.TORCH_DTYPE.get(self.attributes.torch_dtype),
+            dtype=self.TORCH_DTYPE.get(self.attributes.torch_dtype),
             cache_dir=self.attributes.model_cache_dir,
         ).to(self.attributes.device)
 
@@ -199,8 +207,10 @@ class BaseDiffusers(Template, ABC):
             output_type="np",
         )
         generated_images = output.images if output_attribute == "images" else output.frames[0]
+        images_as_uint8 = [(image * 255).clip(0, 255).astype(np.uint8) for image in generated_images]
+        del output
 
-        return [(image * 255).clip(0, 255).astype(np.uint8) for image in generated_images]
+        return images_as_uint8
 
     def _update_images_in_container(self, container: DataContainer, new_packets: list[ImagePacket]) -> None:
         """Updates the container with new image packets based on the `overwrite_images` attribute.
@@ -218,17 +228,39 @@ class BaseDiffusers(Template, ABC):
         else:
             container.images.extend(new_packets)
 
-    def _clear_memory(self) -> None:
+    @staticmethod
+    def clear_memory() -> None:
         """Clears memory to free up resources.
 
         This method performs garbage collection and clears GPU memory (if applicable) to prevent memory leaks
         and ensure efficient resource usage.
         """
-        del self.generator
-        if self.attributes.device == "cuda":
+        gc.collect()
+        if torch.cuda.is_available():
             torch.cuda.empty_cache()
-            torch.cuda.ipc_collect()
 
     def reset_state(self, template_name: str | None = None) -> None:
-        self._clear_memory()
-        super().reset_state(template_name)
+        """Releases the pipeline and processor from memory and re-instantiates the template.
+
+        Args:
+            template_name (str | None, optional): The name of the template instance being reset. Defaults to None.
+        """
+        _ = template_name
+
+        if hasattr(self, "pipeline") and self.pipeline is not None:
+            components = self.pipeline.components
+            for component in components.values():
+                if isinstance(component, torch.nn.Module):
+                    component.to("cpu")
+
+            del self.pipeline
+
+        if hasattr(self, "processor"):
+            del self.processor
+
+        if hasattr(self, "generator"):
+            del self.generator
+
+        self.clear_memory()
+        self.initialize()
+        self.logger.info(f"Reset template instance `{self.instance_name}`")

sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/image_to_image_diffusers.py

@@ -8,7 +8,6 @@ import numpy as np
 from diffusers import AutoPipelineForImage2Image
 from PIL import Image
 from sinapsis_core.data_containers.data_packet import DataContainer, ImagePacket
-from sinapsis_core.template_base.base_models import TemplateAttributeType
 
 from sinapsis_huggingface_diffusers.helpers.tags import Tags
 from sinapsis_huggingface_diffusers.templates.base_diffusers import BaseDiffusers
@@ -52,9 +51,13 @@ class ImageToImageDiffusers(BaseDiffusers):
 
     UIProperties = ImageToImageDiffusersUIProperties
 
-    def __init__(self, attributes: TemplateAttributeType) -> None:
-        super().__init__(attributes)
-        self.requires_pil = False
+    def initialize(self) -> None:
+        """Initializes the template's common state for creation or reset.
+
+        This method is called by both `__init__` and `reset_state` to ensure
+        a consistent state. Can be overriden by subclasses for specific behaviour.
+        """
+        super().initialize()
         self.output_attribute: Literal["images", "frames"] = "images"
         self.num_duplicates = self.num_images_per_prompt
 
@@ -67,7 +70,8 @@ class ImageToImageDiffusers(BaseDiffusers):
         """
         return AutoPipelineForImage2Image
 
-    def _convert_image_format(self, image_packet: ImagePacket) -> np.ndarray | Image.Image:
+    @staticmethod
+    def _convert_image_format(image_packet: ImagePacket) -> Image.Image:
         """Converts the input image into the appropriate format for the pipeline.
 
         The format depends on the `requires_pil` attribute:
@@ -78,12 +82,9 @@ class ImageToImageDiffusers(BaseDiffusers):
             image_packet (ImagePacket): The input image packet.
 
         Returns:
-            np.ndarray | Image.Image: The converted image, either as a normalized NumPy array or a
-                PIL Image.
+            Image.Image: The converted image as a PIL Image.
         """
-        if self.requires_pil:
-            return Image.fromarray(image_packet.content)
-        return image_packet.content / 255.0
+        return Image.fromarray(image_packet.content)
 
     def preprocess_inputs(self, image_packet: ImagePacket) -> dict[str, np.ndarray | list[np.ndarray]]:
         """Prepares the input image for the image-to-image pipeline.
@@ -157,6 +158,6 @@ class ImageToImageDiffusers(BaseDiffusers):
         new_packets = [ImagePacket(content=image) for image in all_generated_images]
         processed_packets, _ = self.post_processing_packets(new_packets, old_packets)
         self._update_images_in_container(container, processed_packets)
-        self._clear_memory()
+        self.clear_memory()
 
         return container

sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/image_to_video_gen_xl_diffusers.py

@@ -1,7 +1,8 @@
 # -*- coding: utf-8 -*-
+from typing import Literal
+
 from diffusers import I2VGenXLPipeline
 from pydantic import Field
-from sinapsis_core.template_base.base_models import TemplateAttributeType
 
 from sinapsis_huggingface_diffusers.helpers.tags import Tags
 from sinapsis_huggingface_diffusers.templates.base_diffusers import BaseDiffusersAttributes, BaseGenerationParams
@@ -72,11 +73,15 @@ class ImageToVideoGenXLDiffusers(ImageToImageDiffusers):
     AttributesBaseModel = ImageToVideoGenXLDiffusersAttributes
     UIProperties = ImageToVideoGenXLDiffusersUIProperties
 
-    def __init__(self, attributes: TemplateAttributeType) -> None:
-        super().__init__(attributes)
+    def initialize(self) -> None:
+        """Initializes the template's common state for creation or reset.
+
+        This method is called by both `__init__` and `reset_state` to ensure
+        a consistent state. Can be overriden by subclasses for specific behaviour.
+        """
+        super().initialize()
+        self.output_attribute: Literal["images", "frames"] = "frames"
         self.num_duplicates = self.attributes.generation_params.num_frames
-        self.requires_pil = True
-        self.output_attribute = "frames"
 
     @staticmethod
     def _pipeline_class() -> I2VGenXLPipeline:

sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/inpainting_diffusers.py

@@ -7,7 +7,6 @@ import cv2
 import numpy as np
 from diffusers import AutoPipelineForInpainting
 from sinapsis_core.data_containers.data_packet import ImageAnnotations, ImagePacket
-from sinapsis_core.template_base.base_models import TemplateAttributeType
 
 from sinapsis_huggingface_diffusers.helpers.tags import Tags
 from sinapsis_huggingface_diffusers.templates.base_diffusers import (
@@ -75,8 +74,13 @@ class InpaintingDiffusers(ImageToImageDiffusers):
     UIProperties = InpaintingDiffusersUIProperties
     AttributesBaseModel = InpaintingDiffusersAttributes
 
-    def __init__(self, attributes: TemplateAttributeType) -> None:
-        super().__init__(attributes)
+    def initialize(self) -> None:
+        """Initializes the template's common state for creation or reset.
+
+        This method is called by both `__init__` and `reset_state` to ensure
+        a consistent state. Can be overriden by subclasses for specific behaviour.
+        """
+        super().initialize()
         if self.attributes.preserve_outside_content and self.attributes.dilation_radius is None:
             raise ValueError("Need to specify a dilation_radius if preserve_outside_content=True")
 

sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/text_to_image_diffusers.py

@@ -80,6 +80,6 @@ class TextToImageDiffusers(BaseDiffusers):
         image_packets = [ImagePacket(content=image) for image in generated_images]
         self._set_packet_sources(image_packets)
         self._update_images_in_container(container, image_packets)
-        self._clear_memory()
+        self.clear_memory()
 
         return container

sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/grounding_dino.py

@@ -42,7 +42,7 @@ class GroundingBaseAttributes(TemplateAttributes):
         device (Literal["cuda", "cpu"]): Device to be used for inference.
     """
 
-    model_path: str
+    model_path: Literal["IDEA-Research/grounding-dino-base"] = "IDEA-Research/grounding-dino-base"
     model_cache_dir: str = str(SINAPSIS_CACHE_DIR)
     inference_mode: Literal["object_detection", "zero_shot"]
     threshold: float = 0.25
@@ -111,11 +111,18 @@ class GroundingDINO(Template):
             attributes (dict[str, Any]): Dictionary containing configuration parameters.
         """
         super().__init__(attributes)
-        self.device = self.attributes.device
+        self.initialize()
+
+    def initialize(self) -> None:
+        """Initializes the template's common state for creation or reset.
+
+        This method is called by both `__init__` and `reset_state` to ensure
+        a consistent state. Can be overriden by subclasses for specific behaviour.
+        """
         self.processor = AutoProcessor.from_pretrained(
             self.attributes.model_path, cache_dir=self.attributes.model_cache_dir
         )
-        self.model = self._set_model().to(self.device)
+        self.model = self._set_model().to(self.attributes.device)
         self.max_tokens = self.processor.tokenizer.model_max_length
         self.text_input = self.validate_and_format_text_input(self.attributes.text_input)
 
@@ -171,7 +178,7 @@ class GroundingDINO(Template):
             images=image_packet.content,
             text=self.text_input,
             return_tensors="pt",
-        ).to(self.device)
+        ).to(self.attributes.device)
 
         with torch.no_grad():
             outputs = self.model(**inputs)
@@ -350,19 +357,32 @@ class GroundingDINO(Template):
 
         return container
 
-    def _clear_memory(self) -> None:
+    @staticmethod
+    def clear_memory() -> None:
         """Clears memory to free up resources.
 
         This method performs garbage collection and clears GPU memory (if applicable) to prevent memory leaks
         and ensure efficient resource usage.
         """
-        for child in self.model.children():
-            child.cpu()
-            gc.collect()
-        if self.attributes.device == "cuda":
+        gc.collect()
+        if torch.cuda.is_available():
             torch.cuda.empty_cache()
-            torch.cuda.ipc_collect()
 
     def reset_state(self, template_name: str | None = None) -> None:
-        self._clear_memory()
-        super().reset_state(template_name)
+        """Releases the pipeline and processor from memory and re-instantiates the template.
+
+        Args:
+            template_name (str | None, optional): The name of the template instance being reset. Defaults to None.
+        """
+        _ = template_name
+
+        if hasattr(self, "model") and self.model is not None:
+            self.model.to("cpu")
+            del self.model
+
+        if hasattr(self, "processor"):
+            del self.processor
+
+        self.clear_memory()
+        self.initialize()
+        self.logger.info(f"Reset template instance `{self.instance_name}`")

sinapsis_huggingface_hub/src/sinapsis_huggingface_hub/helpers/tags.py

@@ -0,0 +1,7 @@
+# -*- coding: utf-8 -*-
+from enum import Enum
+
+
+class Tags(Enum):
+    HUGGINGFACE = "huggingface"
+    MODELS = "models"

sinapsis_huggingface_hub/src/sinapsis_huggingface_hub/templates/__init__.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+import importlib
+from typing import Callable
+
+_root_lib_path = "sinapsis_huggingface_hub.templates"
+
+_template_lookup = {
+    "HuggingFaceDownloader": f"{_root_lib_path}.huggingface_downloader",
+}
+
+
+def __getattr__(name: str) -> Callable:
+    if name in _template_lookup:
+        module = importlib.import_module(_template_lookup[name])
+        return getattr(module, name)
+    raise AttributeError(f"template `{name}` not found in {_root_lib_path}")
+
+
+__all__ = list(_template_lookup.keys())

sinapsis_huggingface_hub/src/sinapsis_huggingface_hub/templates/huggingface_downloader.py

@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+
+from typing import Literal
+
+from huggingface_hub import snapshot_download
+from pydantic import Field
+from sinapsis_core.data_containers.data_packet import DataContainer
+from sinapsis_core.template_base import Template
+from sinapsis_core.template_base.base_models import TemplateAttributes, TemplateAttributeType, UIPropertiesMetadata
+from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
+from sinapsis_huggingface_transformers.helpers.tags import Tags
+
+
+class HuggingFaceDownloaderAttributes(TemplateAttributes):
+    """Defines the configuration for downloading a repository snapshot from the Hugging Face Hub.
+
+    Attributes:
+        repo_id (str): The repository ID from the Hub (e.g., "stabilityai/stable-diffusion-2-1").
+        repo_type (Literal["dataset", "space", "model"]): The type of the repository.
+        revision (str | None): The specific model version to use (e.g., a branch name, tag, or commit hash).
+        cache_dir (str): The directory where downloaded files will be cached.
+        allow_patterns (list[str] | str | None): A pattern or list of patterns to specify which files to download.
+        ignore_patterns (list[str] | str | None): A pattern or list of patterns to specify which files to ignore.
+        max_workers (int): The maximum number of threads to use for parallel downloads.
+        force_download (bool): If True, forces the repository to be re-downloaded even if it's already cached.
+        proxies (dict): A dictionary of proxy servers to use for the download.
+        local_files_only (bool): If True, the function will only look for the files in the cache and will not
+            access the network.
+        etag_timeout (float): The timeout in seconds for checking the ETag of the files.
+        resume_download (bool): If True, resumes an interrupted download.
+    """
+
+    repo_id: str
+    repo_type: Literal["dataset", "space", "model"] = "model"
+    revision: str | None = None
+    cache_dir: str = str(SINAPSIS_CACHE_DIR)
+    allow_patterns: list[str] | str | None = None
+    ignore_patterns: list[str] | str | None = None
+    max_workers: int = 8
+    force_download: bool = False
+    proxies: dict = Field(default_factory=dict)
+    local_files_only: bool = False
+    etag_timeout: float = 10.0
+    resume_download: bool = False
+
+
+class HuggingFaceDownloader(Template):
+    """A Sinapsis Template that downloads a repository snapshot from the Hugging Face Hub."""
+
+    AttributesBaseModel = HuggingFaceDownloaderAttributes
+    UIProperties = UIPropertiesMetadata(
+        category="Hugging Face Hub",
+        tags=[Tags.HUGGINGFACE, Tags.MODELS],
+    )
+
+    def __init__(self, attributes: TemplateAttributeType) -> None:
+        super().__init__(attributes)
+        self.logger.info(f"Starting download of {self.attributes.repo_id} to {self.attributes.cache_dir}")
+        self.path = snapshot_download(**self.attributes.model_dump(exclude_none=True, exclude={"metadata"}))
+
+    def execute(self, container: DataContainer) -> DataContainer:
+        """Injects the local directory path of the downloaded repository into the data container.
+
+        Args:
+            container (DataContainer): The data container to which the directory path will be added.
+
+        Returns:
+            DataContainer: The updated data container, now containing the directory path.
+        """
+        self._set_generic_data(container, self.path)
+        return container

sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/image_to_text_transformers.py

@@ -40,6 +40,7 @@ class ImageToTextTransformersAttributes(TransformersBaseAttributes):
             such as `max_new_tokens`.
     """
 
+    model_path: Literal["Qwen/Qwen-Image"] = "Qwen/Qwen-Image"
     inference_kwargs: ImageToTextInferenceKwargs = Field(default_factory=ImageToTextInferenceKwargs)
 
 

sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_base.py

@@ -33,7 +33,7 @@ class PaliGemmaBaseAttributes(TemplateAttributes):
         torch_dtype (Literal["float16", "float32"]): Model precision type. Defaults to float16.
     """
 
-    model_path: str
+    model_path: Literal["google/paligemma2-3b-pt-224"] = "google/paligemma2-3b-pt-224"
     processor_path: str
     model_cache_dir: str = str(SINAPSIS_CACHE_DIR)
     device: Literal["cuda", "cpu"] = "cpu"

sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/speech_to_text_transformers.py

@@ -41,6 +41,7 @@ class SpeechToTextTransformersAttributes(TransformersBaseAttributes):
             such as `return_timestamps`.
     """
 
+    model_path: Literal["openai/whisper-small"] = "openai/whisper-small"
     inference_kwargs: SpeechToTextInferenceKwargs = Field(default_factory=SpeechToTextInferenceKwargs)
 
 

sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/summarization_transformers.py

@@ -39,6 +39,7 @@ class SummarizationTransformersAttributes(TransformersBaseAttributes):
         inference_kwargs: Task-specific parameters for the summarization pipeline,
             such as `clean_up_tokenization_spaces`.
     """
+    model_path: Literal["facebook/bart-large-cnn", "google/pegasus-xsum", "Falconsai/text_summarization", "facebook/bart-large-xsum"] = "facebook/bart-large-cnn"
 
     inference_kwargs: SummarizationInferenceKwargs = Field(default_factory=SummarizationInferenceKwargs)
 

sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/text_to_speech_transformers.py

@@ -29,6 +29,7 @@ class TextToSpeechAttributes(TransformersBaseAttributes):
             input text into smaller chunks. If not specified, sentences will be split
             based on punctuation.
     """
+    model_path: Literal["suno/bark", "suno/bark-small", "ResembleAI/chatterbox"] = "suno/bark"
 
     use_embeddings: bool = False
     sample_rate: int | None = None