bizyengine 1.2.39__py3-none-any.whl → 1.2.41__py3-none-any.whl

bizyengine/bizyair_extras/nodes_wan_video.py

@@ -1,6 +1,42 @@
 from bizyengine.core import BizyAirBaseNode
 
 
+class WanImageToVideo(BizyAirBaseNode):
+    MAX_RESOLUTION = 960
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "positive": ("CONDITIONING",),
+                "negative": ("CONDITIONING",),
+                "vae": ("VAE",),
+                "width": (
+                    "INT",
+                    {"default": 832, "min": 16, "max": s.MAX_RESOLUTION, "step": 16},
+                ),
+                "height": (
+                    "INT",
+                    {"default": 480, "min": 16, "max": s.MAX_RESOLUTION, "step": 16},
+                ),
+                "length": (
+                    "INT",
+                    {"default": 81, "min": 1, "max": s.MAX_RESOLUTION, "step": 4},
+                ),
+                "batch_size": ("INT", {"default": 1, "min": 1, "max": 4096}),
+            },
+            "optional": {
+                "clip_vision_output": ("CLIP_VISION_OUTPUT",),
+                "start_image": ("IMAGE",),
+            },
+        }
+
+    RETURN_TYPES = ("CONDITIONING", "CONDITIONING", "LATENT")
+    RETURN_NAMES = ("positive", "negative", "latent")
+    CATEGORY = "WanI2V"
+    # FUNCTION = "encode"
+
+
 class Wan_Model_Loader(BizyAirBaseNode):
     @classmethod
     def INPUT_TYPES(cls):

bizyengine/bizyair_extras/route_bizyair_tools.py

@@ -8,6 +8,22 @@ import bizyengine.core as bizyair
 from bizyengine.core.common.env_var import BIZYAIR_DEBUG
 from bizyengine.core.data_types import BIZYAIR_TYPE_MAP
 
+# Invert BIZYAIR_TYPE_MAP for reverse lookup
+INVERTED_BIZYAIR_TYPE_MAP = {v: k for k, v in BIZYAIR_TYPE_MAP.items()}
+
+import importlib
+import warnings
+
+try:
+    comfy_nodes = importlib.import_module("nodes")
+except ModuleNotFoundError:
+    warnings.warn("Importing comfyui.nodes failed!")
+    comfy_nodes = type(
+        "nodes",
+        (object,),
+        {"NODE_DISPLAY_NAME_MAPPINGS": {}, "NODE_CLASS_MAPPINGS": {}},
+    )
+
 
 def get_bizyair_display_name(class_type: str) -> str:
     bizyair_cls_prefix = bizyair.nodes_base.PREFIX
@@ -15,35 +31,69 @@ def get_bizyair_display_name(class_type: str) -> str:
     return f"{bizyair_logo}{bizyair_cls_prefix} {bizyair.NODE_DISPLAY_NAME_MAPPINGS.get(class_type, class_type)}"
 
 
-def workflow_convert(inputs: dict):
+def revert_bizyair_display_name(class_type: str) -> str:
+    comfy_class_type = class_type[len(f"{bizyair.nodes_base.PREFIX}_") :]
+    return comfy_nodes.NODE_DISPLAY_NAME_MAPPINGS.get(
+        comfy_class_type, comfy_class_type
+    )
+
+
+def workflow_convert(inputs: dict, comfy2bizyair: bool = True):
     nodes = inputs["nodes"]
     for node in nodes:
         class_type = node["type"]
         node_inputs = node.get("inputs")
         node_outputs = node.get("outputs")
-        bizyair_cls_type = f"{bizyair.nodes_base.PREFIX}_{class_type}"
         is_converted = False
-        if bizyair_cls_type in bizyair.NODE_CLASS_MAPPINGS:
-            node["type"] = bizyair_cls_type
-            display_name = get_bizyair_display_name(class_type)
+
+        # check if the node is a bizyair node
+        new_class_type = class_type
+        if not comfy2bizyair:
+            if (
+                len(class_type) > len(f"{bizyair.nodes_base.PREFIX}_")
+                and class_type[len(f"{bizyair.nodes_base.PREFIX}_") :]
+                in comfy_nodes.NODE_CLASS_MAPPINGS
+            ):
+                new_class_type = class_type[len(f"{bizyair.nodes_base.PREFIX}_") :]
+        elif (
+            comfy2bizyair
+            and f"{bizyair.nodes_base.PREFIX}_{class_type}"
+            in bizyair.NODE_CLASS_MAPPINGS
+        ):
+            new_class_type = f"{bizyair.nodes_base.PREFIX}_{class_type}"
+
+        if new_class_type != class_type:
+            node["type"] = new_class_type
+            display_name = (
+                get_bizyair_display_name(class_type)
+                if comfy2bizyair
+                else revert_bizyair_display_name(class_type)
+            )
             node["properties"]["Node name for S&R"] = display_name
+
+            type_mapping = (
+                BIZYAIR_TYPE_MAP if comfy2bizyair else INVERTED_BIZYAIR_TYPE_MAP
+            )
             if node_inputs:
                 for input_node in node_inputs:
                     input_type = input_node["type"]
-                    input_node["type"] = BIZYAIR_TYPE_MAP.get(input_type, input_type)
+                    input_node["type"] = type_mapping.get(input_type, input_type)
             if node_outputs:
                 for output_node in node_outputs:
                     output_type = output_node["type"]
-                    output_node["type"] = BIZYAIR_TYPE_MAP.get(output_type, output_type)
+                    output_node["type"] = type_mapping.get(output_type, output_type)
+
             is_converted = True
+
         if BIZYAIR_DEBUG:
             pprint.pprint(
                 {
                     "original_class_type": class_type,
-                    "bizyair_cls_type": bizyair_cls_type,
+                    "new_class_type": new_class_type,
                     "is_converted": is_converted,
                 }
             )
+
     return inputs
 
 
@@ -51,7 +101,13 @@ def workflow_convert(inputs: dict):
 async def convert(request):
     try:
         data = await request.json()
-        ret = workflow_convert(data)
+        comfy2bizyair = data.get("comfy2bizyair", True)
+        ret = workflow_convert(data, comfy2bizyair=comfy2bizyair)
+
+        # remove "comfy2bizyair" key from the response
+        if "comfy2bizyair" in ret:
+            del ret["comfy2bizyair"]
+
         return web.Response(
             text=json.dumps(ret),
             content_type="application/json",

bizyengine/core/commands/servers/prompt_server.py

@@ -189,12 +189,10 @@ class PromptServer(Command):
             and result.get("query_url", None) is not None
         )
 
-    def _get_fass_result(self, result: Dict[str, Any]) -> Dict[str, Any]:
+    def _get_fass_result(self, result: Dict[str, Any], **kwargs) -> Dict[str, Any]:
         request_id = result.get("request_id", None)
         query_url = f"{BIZYAIR_SERVER_ADDRESS}/supernode/status?request_id={request_id}"
-        return self._poll_fass_for_completion(
-            query_url,
-        )
+        return self._poll_fass_for_completion(query_url, **kwargs)
 
     def _poll_fass_for_completion(self, query_url, **kwargs):
         start_time = time.time()
@@ -228,7 +226,7 @@ class PromptServer(Command):
         try:
             if self.is_fass_task(result):
                 self.cache_manager.set(cache_key, result)
-                result = self._get_fass_result(result)
+                result = self._get_fass_result(result, **kwargs)
                 result = result.get("data", {}).get("payload", None)
                 assert result is not None, "Output payload should not be None"
                 self.cache_manager.set(cache_key, result, overwrite=True)

bizyengine/core/configs/models.json

@@ -69,6 +69,8 @@
     "shuttle-3.1-aesthetic.safetensors",
     "wan2.2_t2v_high_noise_14B_fp8_scaled.safetensors",
     "wan2.2_t2v_low_noise_14B_fp8_scaled.safetensors",
+    "wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors",
+    "wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors",
     "flux1-dev-kontext_fp8_scaled.safetensors",
     "flux1-dev-kontext-onediff.safetensors",
     {

bizyengine/core/configs/models.yaml

@@ -447,6 +447,8 @@ model_rules:
         inputs:
           unet_name:
             - ^wan2.2_t2v_high_noise_14B_fp8_scaled.safetensors$
+            - ^wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors$
+            - ^wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors$
 
   - mode_type: Wan2.2-T2V
     base_model: Wan

bizyengine/version.txt

@@ -1 +1 @@
-1.2.39
+1.2.41

{bizyengine-1.2.39.dist-info → bizyengine-1.2.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizyengine
-Version: 1.2.39
+Version: 1.2.41
 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.39.dist-info → bizyengine-1.2.41.dist-info}/RECORD

@@ -1,5 +1,5 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=9oNira_RnApeSDu5Wu73K540tdB-LBj2UA4KTjIiSMM,7
+bizyengine/version.txt,sha256=14oMCP7qyokUFWpNMA8OF-sbwiuToPwd3e8z94cw7ZM,7
 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=1UiFmE2U7r7hCHgsw4-p_YL0VCmTJc9NyYDEbhkanaY,16336
@@ -39,8 +39,8 @@ bizyengine/bizyair_extras/nodes_ultimatesdupscale.py,sha256=-_SsLTAWAQDv4uw-4Z7I
 bizyengine/bizyair_extras/nodes_upscale_model.py,sha256=lrzA1BFI2w5aEPCmNPMh07s-WDzG-xTT49uU6WCnlP8,1151
 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=w04QUWFnavT1zu8ht7Z1WTMlSnn__cO3r5RmiKlcuUQ,2191
+bizyengine/bizyair_extras/nodes_wan_video.py,sha256=aE2JBF0ZT-6BOM0bGu9R4yZ_eMeMnnjCW-YzFe4qg8Q,2804
+bizyengine/bizyair_extras/route_bizyair_tools.py,sha256=EiP5pS6xoE3tULoNSN2hYZA29vgt7yCErsbRp34gGEg,3898
 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
@@ -56,15 +56,15 @@ bizyengine/core/commands/invoker.py,sha256=8wcIMd8k44o96LAvxFrIiKOlVtf1MW-AcMDXs
 bizyengine/core/commands/processors/model_hosting_processor.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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=0QjpAB1JG90yq4a8zvzPtUfqALd5J7Mq69W4I4Ma6xQ,10866
+bizyengine/core/commands/servers/prompt_server.py,sha256=e8JhtKRM8nw0kQwe2Ofl-zQtiVqQdbbWRxOqkFmSclM,10873
 bizyengine/core/common/__init__.py,sha256=GicZw6YeAZk1PsKmFDt9dm1F75zPUlpia9Q_ki5vW1Y,179
 bizyengine/core/common/caching.py,sha256=hRNsSrfNxgc1zzvBzLVjMY0iMkKqA0TBCr-iYhEpzik,6946
 bizyengine/core/common/client.py,sha256=mP24Hxe_y20VXxVFdUz6HtLBzATUS4d8nr-rNf2KNwc,10535
 bizyengine/core/common/env_var.py,sha256=1EAW3gOXY2bKouCqrGa583vTJRdDasQ1IsFTnzDg7Dk,3450
 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=rmH1BQp-j4nWiSo22QJhGIIb3y_Yycfmpkhseq6dgD8,3190
-bizyengine/core/configs/models.yaml,sha256=SgCyNGE-ia5GDgROF9ATDE6YdLYzqoNuVsKP0XjXJ8w,11442
+bizyengine/core/configs/models.json,sha256=v-PpSxNuexSPAVlZbXJyn934cfzoaTC76whFSidaRks,3301
+bizyengine/core/configs/models.yaml,sha256=Zv4H0DXlE1MBKvCFvOQrFjgYdpX1nUZIoNzS7HnbPa4,11571
 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
@@ -79,7 +79,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.39.dist-info/METADATA,sha256=Bp5tyJOcAiD4ubGW7LloYfZqK3ft46RxeYyiGh3CJ8M,675
-bizyengine-1.2.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.39.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.39.dist-info/RECORD,,
+bizyengine-1.2.41.dist-info/METADATA,sha256=H3M0yCeP_Gz6s_izao-RYYhtfwhkrV8jv-UkJqZs-8w,675
+bizyengine-1.2.41.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.41.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.41.dist-info/RECORD,,