neuralnode 2.1.10__tar.gz → 2.1.11__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuralnode
-Version: 2.1.10
+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.10 → neuralnode-2.1.11}/pyproject.toml

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

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

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

@@ -661,6 +661,16 @@ class HorusProvider(BaseLLMProvider):
                             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,
@@ -843,6 +853,16 @@ class HorusProvider(BaseLLMProvider):
                     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(