neuralnode 2.1.3__tar.gz → 2.1.4__tar.gz

{neuralnode-2.1.3 → neuralnode-2.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuralnode
-Version: 2.1.3
+Version: 2.1.4
 Summary: Comprehensive AI Framework with 50+ LLM Providers, Advanced Agents, Chains, Memory, RAG, and 100+ Tools
 Project-URL: Homepage, https://assem.cloud/
 Project-URL: Documentation, https://neuralnode.readthedocs.io

{neuralnode-2.1.3 → neuralnode-2.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "neuralnode"
-version = "2.1.3"
+version = "2.1.4"
 description = "Comprehensive AI Framework with 50+ LLM Providers, Advanced Agents, Chains, Memory, RAG, and 100+ Tools"
 readme = "README.md"
 requires-python = ">=3.9"

{neuralnode-2.1.3 → neuralnode-2.1.4}/src/neuralnode/__init__.py

@@ -42,7 +42,7 @@ Quick Start::
     text = sr.listen()
 """
 
-__version__ = "2.1.3"
+__version__ = "2.1.4"
 __author__ = "NeuralNode Contributors"
 
 # ── Core types ────────────────────────────────────────────────────────────────

{neuralnode-2.1.3 → neuralnode-2.1.4}/src/neuralnode/providers/horus.py

@@ -553,7 +553,41 @@ class HorusProvider(BaseLLMProvider):
             pipeline_kwargs["torch_dtype"] = self.torch_dtype
 
         try:
-            self.model = DiffusionPipeline.from_pretrained(repo_id, **pipeline_kwargs)
+            # Check if this model is Horus-Lens-1.0 which requires custom component mapping
+            if "horus-lens-1.0" in repo_id.lower():
+                try:
+                    try:
+                        from diffusers import ZImagePipeline
+                    except ImportError:
+                        from diffusers.pipelines.z_image.pipeline_z_image import ZImagePipeline
+
+                    class PatchedZImagePipeline(ZImagePipeline):
+                        def __init__(
+                            self,
+                            horus_scheduler,
+                            horus_text_encoder,
+                            horus_tokenizer,
+                            horus_transformer,
+                            horus_vae,
+                        ):
+                            super().__init__(
+                                scheduler=horus_scheduler,
+                                text_encoder=horus_text_encoder,
+                                tokenizer=horus_tokenizer,
+                                transformer=horus_transformer,
+                                vae=horus_vae,
+                            )
+                    
+                    self.model = DiffusionPipeline.from_pretrained(
+                        repo_id,
+                        pipeline_class=PatchedZImagePipeline,
+                        **pipeline_kwargs
+                    )
+                except Exception:
+                    # Fallback to default load if patch fails
+                    self.model = DiffusionPipeline.from_pretrained(repo_id, **pipeline_kwargs)
+            else:
+                self.model = DiffusionPipeline.from_pretrained(repo_id, **pipeline_kwargs)
         except Exception as exc:
             raise RuntimeError(
                 f"Failed to load Horus Lens from '{repo_id}'. "