add mochi nodes

bizyair_extras/__init__.py

@@ -19,3 +19,4 @@
 from .nodes_trellis import *
 from .nodes_ultimatesdupscale import *
 from .nodes_upscale_model import *
+from .nodes_mochi import *

bizyair_extras/nodes_mochi.py

@@ -0,0 +1,22 @@
+from bizyair import BizyAirBaseNode
+
+class BizyAir_mochi(BizyAirBaseNode):
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "prompt": ("STRING", {"default": "Close-up of a chameleon's eye, with its scaly skin changing color. Ultra high resolution 4k."}),
+                "steps": ("INT", {"default": 20, "min": 1, "max": 50}),
+                "width": ("INT", {"default": 480, "min": 64, "max": 1536, "step": 8}),
+                "height": ("INT", {"default": 848, "min": 64, "max": 1536, "step": 8}),
+                "frames": ("INT", {"default": 96, "min": 1, "max": 4096}),
+                "fps": ("INT", {"default": 22, "min": 8, "max": 64}),
+                "filename": ("STRING", {"default": "mochi"}),                    
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("path",)
+    CATEGORY = "☁️BizyAir/Trellis"
+    NODE_DISPLAY_NAME = "☁️BizyAir mochi"
+    OUTPUT_NODE = True

src/bizyair/configs/models.yaml

@@ -307,3 +307,11 @@ model_rules:
     route: /supernode/bizyair-janus-pro-7b
     nodes:
       - class_type: 'JanusModelLoader'
+
+  - mode_type: mochi
+    base_model: mochi
+    describe: mochi
+    score: 1
+    route: /supernode/mochi
+    nodes:
+      - class_type: 'mochi'