bizyengine 1.2.30__py3-none-any.whl → 1.2.32__py3-none-any.whl

bizyengine/bizyair_extras/nodes_nunchaku.py

@@ -11,7 +11,11 @@ class NunchakuFluxDiTLoader(BizyAirBaseNode):
         return {
             "required": {
                 "model_path": (
-                    ["svdq-int4-flux.1-dev", "svdq-int4-flux.1-fill-dev"],
+                    [
+                        "svdq-int4-flux.1-dev",
+                        "svdq-int4-flux.1-fill-dev",
+                        "svdq-int4_r32-flux.1-kontext-dev.safetensors",
+                    ],
                     {"tooltip": "The SVDQuant quantized FLUX.1 models."},
                 ),
                 "cache_threshold": (

bizyengine/bizyair_extras/route_bizyair_tools.py

@@ -5,6 +5,7 @@ from aiohttp import web
 from server import PromptServer
 
 import bizyengine.core as bizyair
+from bizyengine.core.common.env_var import BIZYAIR_DEBUG
 from bizyengine.core.data_types import BIZYAIR_TYPE_MAP
 
 
@@ -35,13 +36,14 @@ def workflow_convert(inputs: dict):
                     output_type = output_node["type"]
                     output_node["type"] = BIZYAIR_TYPE_MAP.get(output_type, output_type)
             is_converted = True
-        pprint.pprint(
-            {
-                "original_class_type": class_type,
-                "bizyair_cls_type": bizyair_cls_type,
-                "is_converted": is_converted,
-            }
-        )
+        if BIZYAIR_DEBUG:
+            pprint.pprint(
+                {
+                    "original_class_type": class_type,
+                    "bizyair_cls_type": bizyair_cls_type,
+                    "is_converted": is_converted,
+                }
+            )
     return inputs
 
 

bizyengine/core/commands/processors/prompt_processor.py

@@ -1,5 +1,7 @@
 import json
+import time
 from collections import deque
+from pathlib import Path
 from typing import Any, Dict, List
 
 from bizyengine.core.commands.base import Processor  # type: ignore
@@ -109,17 +111,28 @@ class PromptProcessor(Processor):
         last_node_ids: List[str],
         **kwargs,
     ):
-        dict = {
+        request_data = {
             "prompt": prompt,
             "last_node_id": last_node_ids[0],
             "exec_info": self._exec_info(prompt, kwargs["api_key"]),
         }
         if "prompt_id" in kwargs:
-            dict["prompt_id"] = kwargs["prompt_id"]
-
+            request_data["prompt_id"] = kwargs["prompt_id"]
+
+        if BIZYAIR_DEBUG:
+            save_dir = Path("runs") / "request_bodies"
+            save_dir.mkdir(parents=True, exist_ok=True)
+            timestamp = time.strftime("%Y%m%d_%H%M%S")
+            prompt_id = request_data.get("prompt_id", "no_prompt_id")
+            file_path = save_dir / f"request_{timestamp}_{prompt_id}.json"
+            file_path.write_text(
+                json.dumps(request_data, indent=2, ensure_ascii=False), encoding="utf-8"
+            )
+
+        json_payload = json.dumps(request_data).encode("utf-8")
         return client.send_request(
             url=url,
-            data=json.dumps(dict).encode("utf-8"),
+            data=json_payload,
             headers=client.headers(api_key=kwargs["api_key"]),
         )
 

bizyengine/core/configs/models.yaml

@@ -62,6 +62,17 @@ model_rules:
           unet_name:
             - ^flux1-dev-kontext-onediff.safetensors$
 
+  - mode_type: unet
+    base_model: FLUX-kontext
+    describe: flux1-kontext
+    score: 3
+    route: /supernode/bizyair-comfyui-flux1-kontext-svdq
+    nodes:
+      - class_type: NunchakuFluxDiTLoader
+        inputs:
+          model_path:
+            - ^svdq-int4_r32-flux.1-kontext-dev.safetensors$
+
   - mode_type: unet
     base_model: FLUX-kontext
     describe: flux1-kontext for bizybot only
@@ -102,6 +113,7 @@ model_rules:
         inputs:
           model_path:
             - ^svdq-int4-flux.1-fill-dev$
+
   - mode_type: unet
     base_model: FLUX
     describe: NunchakuPulidLoader

bizyengine/version.txt

@@ -1 +1 @@
-1.2.30
+1.2.32

{bizyengine-1.2.30.dist-info → bizyengine-1.2.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizyengine
-Version: 1.2.30
+Version: 1.2.32
 Summary: [a/BizyAir](https://github.com/siliconflow/BizyAir) Comfy Nodes that can run in any environment.
 Author-email: SiliconFlow <yaochi@siliconflow.cn>
 Project-URL: Repository, https://github.com/siliconflow/BizyAir

{bizyengine-1.2.30.dist-info → bizyengine-1.2.32.dist-info}/RECORD

@@ -1,5 +1,5 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=JlhCJgvOLCf0knGqzSmKCYF_ZcqtpjyEa5v066aASRM,6
+bizyengine/version.txt,sha256=yc10PMdAsRPIE3P3QW2RFxGobS76YW2vkxyYMV4CLRw,6
 bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
 bizyengine/bizy_server/api_client.py,sha256=t6cwob7hs993oy5WdFcjKtIzfi3S_eUhODdDVv_Hedo,43822
 bizyengine/bizy_server/errno.py,sha256=Q-U96XnZQCuPH_44Om8wnc2-Kh7qFqwLKtox27msU54,16095
@@ -27,7 +27,7 @@ bizyengine/bizyair_extras/nodes_image_utils.py,sha256=BldF_CKD2M01K8-SnG-QV86u3H
 bizyengine/bizyair_extras/nodes_ip2p.py,sha256=GSEFJvrs4f2tv0xwYkWqc8uhsXrzAJVPvvwcw0gTjR0,619
 bizyengine/bizyair_extras/nodes_janus_pro.py,sha256=hAdMsS09RkRHZn9cNwpmyOaH7ODOMjVt9SbBsD-UvbM,2665
 bizyengine/bizyair_extras/nodes_model_advanced.py,sha256=RR2pzvlNW7NEcgtRcQSLZ8Vy7_ygA0NOZDjd7ZfzX5k,1756
-bizyengine/bizyair_extras/nodes_nunchaku.py,sha256=ERxDOQAV-ejgbeRMOCIaPBiSfNpP4G2jmdMNbO-U6hI,8079
+bizyengine/bizyair_extras/nodes_nunchaku.py,sha256=E0E8f9LmEzfn4a_LdWIUBatV9n-8RFLSYlPb90SKKxs,8222
 bizyengine/bizyair_extras/nodes_sd3.py,sha256=lZCxj0IFmuxk1fZTDcRKgVV5QWHjkUdpR4w9-DZbMf4,1727
 bizyengine/bizyair_extras/nodes_segment_anything.py,sha256=x1ei2UggHnB8T6aUtK_ZcUehMALEyLUnDoD5SNJCbFU,7249
 bizyengine/bizyair_extras/nodes_segment_anything_utils.py,sha256=ZefAqrFrevDH3XY_wipr_VwKfeXrgpZEUFaqg_JGOdU,4714
@@ -38,7 +38,7 @@ bizyengine/bizyair_extras/nodes_upscale_model.py,sha256=lrzA1BFI2w5aEPCmNPMh07s-
 bizyengine/bizyair_extras/nodes_utils.py,sha256=whog_tmV-q7JvLEdb03JL3KKsC7wKe3kImzx_jPaQD8,2613
 bizyengine/bizyair_extras/nodes_wan_i2v.py,sha256=3XwcxLHmgrihgXDEzcVOjU6VjqnZa3mErINlY014PFA,8435
 bizyengine/bizyair_extras/nodes_wan_video.py,sha256=d1mCcW9jCj-5Oymmymy0Vz-nwWv36FMGE5Gn-E7Rul4,1632
-bizyengine/bizyair_extras/route_bizyair_tools.py,sha256=DMVc7J0oT3H_cdAzoaog-ulhE7zj__w1t-HlWnIpCg0,2080
+bizyengine/bizyair_extras/route_bizyair_tools.py,sha256=w04QUWFnavT1zu8ht7Z1WTMlSnn__cO3r5RmiKlcuUQ,2191
 bizyengine/bizyair_extras/nodes_ipadapter_plus/__init__.py,sha256=ECKATm_EKi_4G47-FJI4-3rHO3iiF9FVakfSTE-pooE,36
 bizyengine/bizyair_extras/nodes_ipadapter_plus/nodes_ipadapter_plus.py,sha256=lOKRem7oiPs8ZkA_p68HxagAgiCSvn3Rk-L4fSXIjyE,54846
 bizyengine/bizyair_extras/nodes_kolors_mz/__init__.py,sha256=HsCCCphW8q0SrWEiFlZKK_W2lQr1T0UJIJL7gEn37ME,3729
@@ -52,7 +52,7 @@ bizyengine/core/commands/__init__.py,sha256=82yRdMT23RTiZPkFW_G3fVa-fj3-TUAXnj6c
 bizyengine/core/commands/base.py,sha256=TYH9lhr033B2roBLPkWkxcvoz_fW3cdzx_bvP_FI7dg,635
 bizyengine/core/commands/invoker.py,sha256=8wcIMd8k44o96LAvxFrIiKOlVtf1MW-AcMDXsZkb5Qc,215
 bizyengine/core/commands/processors/model_hosting_processor.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bizyengine/core/commands/processors/prompt_processor.py,sha256=RpNFzlFwpf_PsgmKpIcuqg9eTOm-EB_sNsyfyRDBaho,4391
+bizyengine/core/commands/processors/prompt_processor.py,sha256=jG0EuphvtycH5Dm-frBGtlIcxkTCXiFMQ1vFMvi_Re0,4963
 bizyengine/core/commands/servers/model_server.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bizyengine/core/commands/servers/prompt_server.py,sha256=f5qTTyoe7_cFJ8xESG--mvP2axATKUgHyTBeV6JrBCI,8529
 bizyengine/core/common/__init__.py,sha256=GicZw6YeAZk1PsKmFDt9dm1F75zPUlpia9Q_ki5vW1Y,179
@@ -62,7 +62,7 @@ bizyengine/core/common/env_var.py,sha256=1EAW3gOXY2bKouCqrGa583vTJRdDasQ1IsFTnzD
 bizyengine/core/common/utils.py,sha256=bm-XmSPy83AyjD0v5EfWp6jiO6_5p7rkZ_HQAuVmgmo,3086
 bizyengine/core/configs/conf.py,sha256=D_UWG9SSJnK5EhbrfNFryJQ8hUwwdvhOGlq1TielwpI,3830
 bizyengine/core/configs/models.json,sha256=NqYVmp-ICpr1vFgafWjJMu5dyuRt9ffVUJmov3_-xRY,2746
-bizyengine/core/configs/models.yaml,sha256=K6b6eNMZCM_bHWBME8DlF2FyonD_9qXLBHcxKiP6WYg,9859
+bizyengine/core/configs/models.yaml,sha256=vK8knwfQm9RS4y5NXrDp2tt76-x08gFHxy7cnC7wA5w,10160
 bizyengine/core/path_utils/__init__.py,sha256=JVpqNHgaKiEtTI8_r47af8GtWHxrtOockQ6Qpzp9_MQ,260
 bizyengine/core/path_utils/path_manager.py,sha256=tRVAcpsYvfWD-tK7khLvNCZayB0wpU9L0tRTH4ZESzM,10549
 bizyengine/core/path_utils/utils.py,sha256=kQfPQjGU27qF9iyzRxLSRg5cMsd-VixuCUldART7cgY,2394
@@ -77,7 +77,7 @@ bizyengine/misc/route_sam.py,sha256=-bMIR2QalfnszipGxSxvDAHGJa5gPSrjkYPb5baaRg4,
 bizyengine/misc/segment_anything.py,sha256=wNKYwlYPMszfwj23524geFZJjZaG4eye65SGaUnh77I,8941
 bizyengine/misc/supernode.py,sha256=STN9gaxfTSErH8OiHeZa47d8z-G9S0I7fXuJvHQOBFM,4532
 bizyengine/misc/utils.py,sha256=deQjBgLAkxIr-NaOMm77TcgBT3ExAp0MFm5ehUJ3CGs,6829
-bizyengine-1.2.30.dist-info/METADATA,sha256=Sk1xkJ14fUPjoOi-V_qMnI07h-uQGMVp5gKarw8uVTg,675
-bizyengine-1.2.30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.30.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.30.dist-info/RECORD,,
+bizyengine-1.2.32.dist-info/METADATA,sha256=I3PGMY7lLy3N6D9-Q7jPPZIlaT9VI7zK7QbuXUpYXlg,675
+bizyengine-1.2.32.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.32.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.32.dist-info/RECORD,,