neuralnode 2.1.9__tar.gz → 2.1.11__tar.gz

{neuralnode-2.1.9 → neuralnode-2.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuralnode
-Version: 2.1.9
+Version: 2.1.11
 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.9 → neuralnode-2.1.11}/publish.bat

@@ -47,7 +47,7 @@ twine check dist/*
 
 REM Upload to PyPI
 echo 📤 Uploading to PyPI...
-twine upload dist/* --username __token__ --password pypi-AgEIcHlwaS5vcmcCJGQ1ZjQ2NDQyLThjNjctNGNlYi1hOTlhLTYyMGFlZGQ2MzAxOAACKlszLCJiMzI5ODAyMi04MjlhLTRlODItOWQzMC1iNzlmZWM2NjdlZmIiXQAABiDYVt0Dx5n9hI2ZtbQxW1v_fvfgFJd_XwhC3ZT9hp6wfw
+twine upload dist/* --username __token__ --password pypi-AgEIcHlwaS5vcmcCJGNhOWI0OTJlLWRkNTAtNGU1OC04MDBkLTVjZDVjNzg2MDc0ZAACKlszLCJiMzI5ODAyMi04MjlhLTRlODItOWQzMC1iNzlmZWM2NjdlZmIiXQAABiB05_WnyXXN1F0zHr2jnG3HZt5Nir65IqELsuEDa71XZQ
 
 REM Verify upload
 echo ✅ Verifying upload...

{neuralnode-2.1.9 → neuralnode-2.1.11}/publish.sh

@@ -48,7 +48,7 @@ twine check dist/*
 echo "📤 Uploading to PyPI..."
 twine upload dist/* \
     --username __token__ \
-    --password pypi-AgEIcHlwaS5vcmcCJGQ1ZjQ2NDQyLThjNjctNGNlYi1hOTlhLTYyMGFlZGQ2MzAxOAACKlszLCJiMzI5ODAyMi04MjlhLTRlODItOWQzMC1iNzlmZWM2NjdlZmIiXQAABiDYVt0Dx5n9hI2ZtbQxW1v_fvfgFJd_XwhC3ZT9hp6wfw
+    --password pypi-AgEIcHlwaS5vcmcCJGNhOWI0OTJlLWRkNTAtNGU1OC04MDBkLTVjZDVjNzg2MDc0ZAACKlszLCJiMzI5ODAyMi04MjlhLTRlODItOWQzMC1iNzlmZWM2NjdlZmIiXQAABiB05_WnyXXN1F0zHr2jnG3HZt5Nir65IqELsuEDa71XZQ
 
 # Verify upload
 echo "✅ Verifying upload..."

{neuralnode-2.1.9 → neuralnode-2.1.11}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "neuralnode"
-version = "2.1.9"
+version = "2.1.11"
 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.9 → neuralnode-2.1.11}/src/neuralnode/__init__.py

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

{neuralnode-2.1.9 → neuralnode-2.1.11}/src/neuralnode/providers/horus.py

@@ -647,13 +647,30 @@ class HorusProvider(BaseLLMProvider):
                             horus_transformer,
                             horus_vae,
                         ):
-                            super().__init__(
-                                scheduler=horus_scheduler,
-                                text_encoder=horus_text_encoder,
-                                tokenizer=horus_tokenizer,
-                                transformer=horus_transformer,
-                                vae=horus_vae,
+                            from diffusers import DiffusionPipeline
+                            DiffusionPipeline.__init__(self)
+                            self.register_modules(
+                                horus_scheduler=horus_scheduler,
+                                horus_text_encoder=horus_text_encoder,
+                                horus_tokenizer=horus_tokenizer,
+                                horus_transformer=horus_transformer,
+                                horus_vae=horus_vae,
                             )
+                            self.scheduler = horus_scheduler
+                            self.text_encoder = horus_text_encoder
+                            self.tokenizer = horus_tokenizer
+                            self.transformer = horus_transformer
+                            self.vae = horus_vae
+
+                            # Initialize VAE scale factor and image processor
+                            vae_scale = 8
+                            if hasattr(self, "vae") and self.vae is not None:
+                                if hasattr(self.vae, "config") and getattr(self.vae.config, "block_out_channels", None) is not None:
+                                    vae_scale = 2 ** (len(self.vae.config.block_out_channels) - 1)
+                            self.vae_scale_factor = vae_scale
+                            
+                            from diffusers.image_processor import VaeImageProcessor
+                            self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor * 2)
                     
                     self.model = PatchedZImagePipeline.from_pretrained(
                         repo_id,
@@ -822,13 +839,30 @@ class HorusProvider(BaseLLMProvider):
                     horus_transformer,
                     horus_vae,
                 ):
-                    super().__init__(
-                        scheduler=horus_scheduler,
-                        text_encoder=horus_text_encoder,
-                        tokenizer=horus_tokenizer,
-                        transformer=horus_transformer,
-                        vae=horus_vae,
+                    from diffusers import DiffusionPipeline
+                    DiffusionPipeline.__init__(self)
+                    self.register_modules(
+                        horus_scheduler=horus_scheduler,
+                        horus_text_encoder=horus_text_encoder,
+                        horus_tokenizer=horus_tokenizer,
+                        horus_transformer=horus_transformer,
+                        horus_vae=horus_vae,
                     )
+                    self.scheduler = horus_scheduler
+                    self.text_encoder = horus_text_encoder
+                    self.tokenizer = horus_tokenizer
+                    self.transformer = horus_transformer
+                    self.vae = horus_vae
+
+                    # Initialize VAE scale factor and image processor
+                    vae_scale = 8
+                    if hasattr(self, "vae") and self.vae is not None:
+                        if hasattr(self.vae, "config") and getattr(self.vae.config, "block_out_channels", None) is not None:
+                            vae_scale = 2 ** (len(self.vae.config.block_out_channels) - 1)
+                    self.vae_scale_factor = vae_scale
+                    
+                    from diffusers.image_processor import VaeImageProcessor
+                    self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor * 2)
             
             with self._quiet_native_output():
                 self.model = PatchedZImagePipeline.from_pretrained(