Add flux1 dev pixel wave (#205)

* add support for PixelWave flux * refine * refine * refine
siliconflow · Nov 6, 2024 · 7a784a1 · 7a784a1
1 parent 5300f4b
commit 7a784a1
Show file tree

Hide file tree

Showing 6 changed files with 750 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,5 @@
 # BizyAir
+- [2024/11/06] 🌩️ BizyAir PixelWave Flux.1-dev Text to Image node is released. [PixelWave Flux.1-dev Text to Image](./examples/bizyair_flux_pixelwave_txt2img.json)
 - [2024/10/30] 🌩️ BizyAir Segment-Anything nodes has been released. [text_guided_segment-anything](./examples/bizyair_text_guided_segment-anything.json) and [box_point_guided_segment-anything](./examples/bizyair_box_point_guided_segment-anything.json)
 - [2024/10/24] 🌩️ BizyAir supports stable Diffusion 3.5. [Stable Diffusion 3.5 Text to Image](./examples/bizyair_sd3_5_txt2img.json)
 - [2024/10/21] 🌩️ BizyAir supports VLModels. We add a tutorial for utilizing the VLM node. [AI Assistant](https://siliconflow.github.io/BizyAir/ai-assistants/introduce.html)

diff --git a/bizyair_example_menu.json b/bizyair_example_menu.json
@@ -4,7 +4,8 @@
         "FLUX-dev Text to Image": "bizyair_flux_dev_workflow.json",
         "FLUX-dev Image to Image": "bizyair_flux_img2img_workflow.json",
         "FLUX Simple Lora ": "bizyair_flux_simple_lora_workflow.json",
-        "FLUX PuLID": "bizyair_flux_pulid.json"
+        "FLUX PuLID": "bizyair_flux_pulid.json",
+        "PixelWave Flux.1-dev Text to Image": "bizyair_flux_pixelwave_txt2img.json"
     },
     "ControlNet Union": {
         "Generate an image from a line drawing": "bizyair_showcase_interior_design.json",

diff --git a/bizyair_extras/nodes_model_advanced.py b/bizyair_extras/nodes_model_advanced.py
@@ -1,4 +1,5 @@
 # ComfyUI/comfy_extras/nodes_model_advanced.py
+import nodes
 from bizyair import BizyAirBaseNode, BizyAirNodeIO, data_types
 
 
@@ -19,3 +20,43 @@ def INPUT_TYPES(s):
     # FUNCTION = "patch"
 
     CATEGORY = "advanced/model"
+
+
+class ModelSamplingFlux(BizyAirBaseNode):
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "model": (data_types.MODEL,),
+                "max_shift": (
+                    "FLOAT",
+                    {"default": 1.15, "min": 0.0, "max": 100.0, "step": 0.01},
+                ),
+                "base_shift": (
+                    "FLOAT",
+                    {"default": 0.5, "min": 0.0, "max": 100.0, "step": 0.01},
+                ),
+                "width": (
+                    "INT",
+                    {
+                        "default": 1024,
+                        "min": 16,
+                        "max": nodes.MAX_RESOLUTION,
+                        "step": 8,
+                    },
+                ),
+                "height": (
+                    "INT",
+                    {
+                        "default": 1024,
+                        "min": 16,
+                        "max": nodes.MAX_RESOLUTION,
+                        "step": 8,
+                    },
+                ),
+            }
+        }
+
+    RETURN_TYPES = (data_types.MODEL,)
+    # FUNCTION = "patch"
+    CATEGORY = "advanced/model"