bizyengine 1.2.33__py3-none-any.whl → 1.2.34__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -17,6 +17,7 @@ from .nodes_janus_pro import *
17
17
  from .nodes_kolors_mz import *
18
18
  from .nodes_model_advanced import *
19
19
  from .nodes_nunchaku import *
20
+ from .nodes_sam2 import *
20
21
  from .nodes_sd3 import *
21
22
  from .nodes_segment_anything import *
22
23
  from .nodes_testing_utils import *
@@ -0,0 +1,324 @@
1
+ from bizyengine.core import BizyAirBaseNode
2
+
3
+
4
+ # Parse Text to BBox
5
+ class ParseQwenVLBBoxes(BizyAirBaseNode):
6
+ @classmethod
7
+ def INPUT_TYPES(cls):
8
+ return {
9
+ "required": {
10
+ "model": (
11
+ [
12
+ "Qwen/Qwen2.5-VL-7B-Instruct",
13
+ "Qwen/Qwen2.5-VL-32B-Instruct",
14
+ "Qwen/Qwen2.5-VL-72B-Instruct",
15
+ ],
16
+ ),
17
+ "bboxes_text": ("STRING",),
18
+ "image": ("IMAGE",),
19
+ "target": ("STRING",),
20
+ "bbox_selection": (
21
+ "STRING",
22
+ {
23
+ "default": "all",
24
+ "tooltip": "all或者逗号分隔的序列号",
25
+ },
26
+ ),
27
+ "merge_boxes": ("BOOLEAN", {"default": False}),
28
+ }
29
+ }
30
+
31
+ RETURN_TYPES = ("JSON", "BBOXES")
32
+ RETURN_NAMES = ("text", "bboxes")
33
+ CATEGORY = "Qwen2.5-VL"
34
+ NODE_DISPLAY_NAME = "Parse Qwen VL BBoxes Text for SAM2"
35
+
36
+
37
+ # LayerMask Advance plugin
38
+ # class LayerMaskSAM2Ultra(BizyAirBaseNode):
39
+
40
+ # def __init__(self):
41
+ # self.NODE_NAME = "SAM2 Ultra"
42
+ # pass
43
+
44
+ # @classmethod
45
+ # def INPUT_TYPES(cls):
46
+ # sam2_model_list = [
47
+ # "sam2_hiera_base_plus.safetensors",
48
+ # "sam2_hiera_large.safetensors",
49
+ # "sam2_hiera_small.safetensors",
50
+ # "sam2_hiera_tiny.safetensors",
51
+ # "sam2.1_hiera_base_plus.safetensors",
52
+ # "sam2.1_hiera_large.safetensors",
53
+ # "sam2.1_hiera_small.safetensors",
54
+ # "sam2.1_hiera_tiny.safetensors",
55
+ # ]
56
+ # model_precision_list = ["fp16", "bf16", "fp32"]
57
+ # select_list = ["all", "first", "by_index"]
58
+ # method_list = [
59
+ # "VITMatte",
60
+ # "VITMatte(local)",
61
+ # "PyMatting",
62
+ # "GuidedFilter",
63
+ # ]
64
+ # device_list = ["cuda"]
65
+ # return {
66
+ # "required": {
67
+ # "image": ("IMAGE",),
68
+ # "bboxes": ("BBOXES",),
69
+ # "sam2_model": (sam2_model_list,),
70
+ # "precision": (model_precision_list,),
71
+ # "bbox_select": (select_list,),
72
+ # "select_index": (
73
+ # "STRING",
74
+ # {"default": "0,"},
75
+ # ),
76
+ # "cache_model": ("BOOLEAN", {"default": False}),
77
+ # "detail_method": (method_list,),
78
+ # "detail_erode": (
79
+ # "INT",
80
+ # {"default": 6, "min": 1, "max": 255, "step": 1},
81
+ # ),
82
+ # "detail_dilate": (
83
+ # "INT",
84
+ # {"default": 4, "min": 1, "max": 255, "step": 1},
85
+ # ),
86
+ # "black_point": (
87
+ # "FLOAT",
88
+ # {
89
+ # "default": 0.15,
90
+ # "min": 0.01,
91
+ # "max": 0.98,
92
+ # "step": 0.01,
93
+ # "display": "slider",
94
+ # },
95
+ # ),
96
+ # "white_point": (
97
+ # "FLOAT",
98
+ # {
99
+ # "default": 0.99,
100
+ # "min": 0.02,
101
+ # "max": 0.99,
102
+ # "step": 0.01,
103
+ # "display": "slider",
104
+ # },
105
+ # ),
106
+ # "process_detail": ("BOOLEAN", {"default": True}),
107
+ # "device": (device_list,),
108
+ # "max_megapixels": (
109
+ # "FLOAT",
110
+ # {"default": 2.0, "min": 1, "max": 999, "step": 0.1},
111
+ # ),
112
+ # },
113
+ # "optional": {},
114
+ # }
115
+
116
+ # RETURN_TYPES = (
117
+ # "IMAGE",
118
+ # "MASK",
119
+ # )
120
+ # RETURN_NAMES = (
121
+ # "image",
122
+ # "mask",
123
+ # )
124
+ # CATEGORY = "😺dzNodes/LayerMask"
125
+ # NODE_DISPLAY_NAME = "LayerMask: SAM2 Ultra(Advance)"
126
+ # CLASS_TYPE_NAME = "LayerMask: SAM2Ultra"
127
+
128
+
129
+ class LayerMaskLoadSAM2Model(BizyAirBaseNode):
130
+
131
+ def __init__(self):
132
+ self.NODE_NAME = "Load SAM2 Model"
133
+ pass
134
+
135
+ @classmethod
136
+ def INPUT_TYPES(cls):
137
+ sam2_model_list = [
138
+ # "sam2_hiera_base_plus.safetensors",
139
+ # "sam2_hiera_large.safetensors",
140
+ # "sam2_hiera_small.safetensors",
141
+ # "sam2_hiera_tiny.safetensors",
142
+ "sam2.1_hiera_base_plus.safetensors",
143
+ # "sam2.1_hiera_large.safetensors",
144
+ # "sam2.1_hiera_small.safetensors",
145
+ # "sam2.1_hiera_tiny.safetensors",
146
+ ]
147
+ model_precision_list = [
148
+ # "fp16",
149
+ "bf16",
150
+ # "fp32"
151
+ ]
152
+ device_list = ["cuda"]
153
+ return {
154
+ "required": {
155
+ "sam2_model": (sam2_model_list,),
156
+ "precision": (model_precision_list,),
157
+ "device": (device_list,),
158
+ },
159
+ "optional": {},
160
+ }
161
+
162
+ RETURN_TYPES = ("LS_SAM2_MODEL",)
163
+ RETURN_NAMES = ("sam2_model",)
164
+ CATEGORY = "😺dzNodes/LayerMask"
165
+ NODE_DISPLAY_NAME = "LayerMask: Load SAM2 Model(Advance)"
166
+ CLASS_TYPE_NAME = "LayerMask: LoadSAM2Model"
167
+
168
+
169
+ class LayerMaskSAM2UltraV2(BizyAirBaseNode):
170
+
171
+ def __init__(self):
172
+ self.NODE_NAME = "SAM2 Ultra V2"
173
+ pass
174
+
175
+ @classmethod
176
+ def INPUT_TYPES(cls):
177
+
178
+ select_list = ["all", "first", "by_index"]
179
+ method_list = [
180
+ "VITMatte",
181
+ "VITMatte(local)",
182
+ "PyMatting",
183
+ "GuidedFilter",
184
+ ]
185
+ return {
186
+ "required": {
187
+ "sam2_model": ("LS_SAM2_MODEL",),
188
+ "image": ("IMAGE",),
189
+ "bboxes": ("BBOXES",),
190
+ "bbox_select": (select_list,),
191
+ "select_index": (
192
+ "STRING",
193
+ {"default": "0,"},
194
+ ),
195
+ "detail_method": (method_list,),
196
+ "detail_erode": (
197
+ "INT",
198
+ {"default": 6, "min": 1, "max": 255, "step": 1},
199
+ ),
200
+ "detail_dilate": (
201
+ "INT",
202
+ {"default": 4, "min": 1, "max": 255, "step": 1},
203
+ ),
204
+ "black_point": (
205
+ "FLOAT",
206
+ {
207
+ "default": 0.15,
208
+ "min": 0.01,
209
+ "max": 0.98,
210
+ "step": 0.01,
211
+ "display": "slider",
212
+ },
213
+ ),
214
+ "white_point": (
215
+ "FLOAT",
216
+ {
217
+ "default": 0.99,
218
+ "min": 0.02,
219
+ "max": 0.99,
220
+ "step": 0.01,
221
+ "display": "slider",
222
+ },
223
+ ),
224
+ "process_detail": ("BOOLEAN", {"default": True}),
225
+ "max_megapixels": (
226
+ "FLOAT",
227
+ {"default": 2.0, "min": 1, "max": 999, "step": 0.1},
228
+ ),
229
+ },
230
+ "optional": {},
231
+ }
232
+
233
+ RETURN_TYPES = (
234
+ "IMAGE",
235
+ "MASK",
236
+ )
237
+ RETURN_NAMES = (
238
+ "image",
239
+ "mask",
240
+ )
241
+ CATEGORY = "😺dzNodes/LayerMask"
242
+ NODE_DISPLAY_NAME = "LayerMask: SAM2 Ultra V2(Advance)"
243
+ CLASS_TYPE_NAME = "LayerMask: SAM2UltraV2"
244
+
245
+
246
+ # class LayerMaskSAM2VideoUltra(BizyAirBaseNode):
247
+
248
+ # def __init__(self):
249
+ # self.NODE_NAME = "SAM2 Video Ultra"
250
+
251
+ # @classmethod
252
+ # def INPUT_TYPES(cls):
253
+ # sam2_model_list = [
254
+ # "sam2_hiera_base_plus.safetensors",
255
+ # "sam2_hiera_large.safetensors",
256
+ # "sam2_hiera_small.safetensors",
257
+ # "sam2_hiera_tiny.safetensors",
258
+ # "sam2.1_hiera_base_plus.safetensors",
259
+ # "sam2.1_hiera_large.safetensors",
260
+ # "sam2.1_hiera_small.safetensors",
261
+ # "sam2.1_hiera_tiny.safetensors",
262
+ # ]
263
+ # model_precision_list = ["fp16", "bf16"]
264
+ # method_list = ["VITMatte"]
265
+ # device_list = ["cuda"]
266
+ # return {
267
+ # "required": {
268
+ # "image": ("IMAGE",),
269
+ # "sam2_model": (sam2_model_list,),
270
+ # "precision": (model_precision_list,),
271
+ # "cache_model": ("BOOLEAN", {"default": False}),
272
+ # "individual_objects": ("BOOLEAN", {"default": False}),
273
+ # "mask_preview_color": (
274
+ # "STRING",
275
+ # {"default": "#FF0080"},
276
+ # ),
277
+ # "detail_method": (method_list,),
278
+ # "detail_erode": (
279
+ # "INT",
280
+ # {"default": 6, "min": 1, "max": 255, "step": 1},
281
+ # ),
282
+ # "detail_dilate": (
283
+ # "INT",
284
+ # {"default": 4, "min": 1, "max": 255, "step": 1},
285
+ # ),
286
+ # "black_point": (
287
+ # "FLOAT",
288
+ # {
289
+ # "default": 0.15,
290
+ # "min": 0.01,
291
+ # "max": 0.98,
292
+ # "step": 0.01,
293
+ # "display": "slider",
294
+ # },
295
+ # ),
296
+ # "white_point": (
297
+ # "FLOAT",
298
+ # {
299
+ # "default": 0.99,
300
+ # "min": 0.02,
301
+ # "max": 0.99,
302
+ # "step": 0.01,
303
+ # "display": "slider",
304
+ # },
305
+ # ),
306
+ # "process_detail": ("BOOLEAN", {"default": True}),
307
+ # "device": (device_list,),
308
+ # "max_megapixels": (
309
+ # "FLOAT",
310
+ # {"default": 0.5, "min": 0.1, "max": 10, "step": 0.1},
311
+ # ),
312
+ # },
313
+ # "optional": {
314
+ # "bboxes": ("BBOXES",),
315
+ # "first_frame_mask": ("MASK",),
316
+ # "pre_mask": ("MASK",),
317
+ # },
318
+ # }
319
+
320
+ # RETURN_TYPES = ("MASK", "IMAGE")
321
+ # RETURN_NAMES = ("mask", "preview")
322
+ # CATEGORY = "😺dzNodes/LayerMask"
323
+ # NODE_DISPLAY_NAME = "LayerMask: SAM2 Video Ultra(Advance)"
324
+ # CLASS_TYPE_NAME = "LayerMask: SAM2VideoUltra"
@@ -416,3 +416,13 @@ model_rules:
416
416
  route: /supernode/bizyair-comfy-hy3d-2.1
417
417
  nodes:
418
418
  - class_type: Hy3DExportMesh
419
+
420
+ - mode_type: sam2
421
+ base_model: SAM2
422
+ describe: SAM2
423
+ score: 1
424
+ route: /supernode/bizyair-sam2
425
+ nodes:
426
+ - class_type: "LayerMask: SAM2Ultra"
427
+ - class_type: "LayerMask: SAM2UltraV2"
428
+ - class_type: "LayerMask: SAM2VideoUltra"
bizyengine/version.txt CHANGED
@@ -1 +1 @@
1
- 1.2.33
1
+ 1.2.34
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: bizyengine
3
- Version: 1.2.33
3
+ Version: 1.2.34
4
4
  Summary: [a/BizyAir](https://github.com/siliconflow/BizyAir) Comfy Nodes that can run in any environment.
5
5
  Author-email: SiliconFlow <yaochi@siliconflow.cn>
6
6
  Project-URL: Repository, https://github.com/siliconflow/BizyAir
@@ -1,5 +1,5 @@
1
1
  bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
2
- bizyengine/version.txt,sha256=uruE8qHcOMHtyKGjgFCkaQ8HCk39_t-NlbecbI-aF48,7
2
+ bizyengine/version.txt,sha256=jCVGl77SOOnOTi3qVjDq9YwNgovc28doMLbEAdKZfiY,7
3
3
  bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
4
4
  bizyengine/bizy_server/api_client.py,sha256=t6cwob7hs993oy5WdFcjKtIzfi3S_eUhODdDVv_Hedo,43822
5
5
  bizyengine/bizy_server/errno.py,sha256=1UiFmE2U7r7hCHgsw4-p_YL0VCmTJc9NyYDEbhkanaY,16336
@@ -10,7 +10,7 @@ bizyengine/bizy_server/resp.py,sha256=iOFT5Ud7VJBP2uqkojJIgc3y2ifMjjEXoj0ewneL9l
10
10
  bizyengine/bizy_server/server.py,sha256=M9P0oungSiUFGRxev3jJxhUTvjm8ixREQ-DCoyzU0gA,58617
11
11
  bizyengine/bizy_server/stream_response.py,sha256=H2XHqlVRtQMhgdztAuG7l8-iV_Pm42u2x6WJ0gNVIW0,9654
12
12
  bizyengine/bizy_server/utils.py,sha256=dgTyLdJtj3POzIC17TOPx_S7WllRJDdyvF8FdSmBSfU,3957
13
- bizyengine/bizyair_extras/__init__.py,sha256=EjyLEdLpPm5I4Z25YmnoK4PYQDmsEyXtCXQt9UDGmi0,1009
13
+ bizyengine/bizyair_extras/__init__.py,sha256=NKC9MOSeRJW1FrQMf4KXUb8ogwrSGhVyCPA4rSULd-I,1035
14
14
  bizyengine/bizyair_extras/nodes_advanced_refluxcontrol.py,sha256=cecfjrtnjJAty9aNkhz8BlmHUC1NImkFlUDiA0COEa4,2242
15
15
  bizyengine/bizyair_extras/nodes_cogview4.py,sha256=Ni0TDOycczyDhYPvSR68TxGV_wE2uhaxd8MIj-J4-3o,2031
16
16
  bizyengine/bizyair_extras/nodes_comfyui_detail_daemon.py,sha256=i71it24tiGvZ3h-XFWISr4CpZszUtPuz3UrZARYluLk,6169
@@ -28,6 +28,7 @@ bizyengine/bizyair_extras/nodes_ip2p.py,sha256=GSEFJvrs4f2tv0xwYkWqc8uhsXrzAJVPv
28
28
  bizyengine/bizyair_extras/nodes_janus_pro.py,sha256=hAdMsS09RkRHZn9cNwpmyOaH7ODOMjVt9SbBsD-UvbM,2665
29
29
  bizyengine/bizyair_extras/nodes_model_advanced.py,sha256=RR2pzvlNW7NEcgtRcQSLZ8Vy7_ygA0NOZDjd7ZfzX5k,1756
30
30
  bizyengine/bizyair_extras/nodes_nunchaku.py,sha256=E0E8f9LmEzfn4a_LdWIUBatV9n-8RFLSYlPb90SKKxs,8222
31
+ bizyengine/bizyair_extras/nodes_sam2.py,sha256=JTrB7ELwhw6_uOjykRWK9KyOqpYoOtJiGKMxFUbHnNQ,10554
31
32
  bizyengine/bizyair_extras/nodes_sd3.py,sha256=lZCxj0IFmuxk1fZTDcRKgVV5QWHjkUdpR4w9-DZbMf4,1727
32
33
  bizyengine/bizyair_extras/nodes_segment_anything.py,sha256=x1ei2UggHnB8T6aUtK_ZcUehMALEyLUnDoD5SNJCbFU,7249
33
34
  bizyengine/bizyair_extras/nodes_segment_anything_utils.py,sha256=ZefAqrFrevDH3XY_wipr_VwKfeXrgpZEUFaqg_JGOdU,4714
@@ -62,7 +63,7 @@ bizyengine/core/common/env_var.py,sha256=1EAW3gOXY2bKouCqrGa583vTJRdDasQ1IsFTnzD
62
63
  bizyengine/core/common/utils.py,sha256=bm-XmSPy83AyjD0v5EfWp6jiO6_5p7rkZ_HQAuVmgmo,3086
63
64
  bizyengine/core/configs/conf.py,sha256=D_UWG9SSJnK5EhbrfNFryJQ8hUwwdvhOGlq1TielwpI,3830
64
65
  bizyengine/core/configs/models.json,sha256=NqYVmp-ICpr1vFgafWjJMu5dyuRt9ffVUJmov3_-xRY,2746
65
- bizyengine/core/configs/models.yaml,sha256=vK8knwfQm9RS4y5NXrDp2tt76-x08gFHxy7cnC7wA5w,10160
66
+ bizyengine/core/configs/models.yaml,sha256=z51inNcqTWRJsaxyaD7_H11_H5ubc8KWOx5OIXuthdQ,10416
66
67
  bizyengine/core/path_utils/__init__.py,sha256=JVpqNHgaKiEtTI8_r47af8GtWHxrtOockQ6Qpzp9_MQ,260
67
68
  bizyengine/core/path_utils/path_manager.py,sha256=tRVAcpsYvfWD-tK7khLvNCZayB0wpU9L0tRTH4ZESzM,10549
68
69
  bizyengine/core/path_utils/utils.py,sha256=kQfPQjGU27qF9iyzRxLSRg5cMsd-VixuCUldART7cgY,2394
@@ -77,7 +78,7 @@ bizyengine/misc/route_sam.py,sha256=-bMIR2QalfnszipGxSxvDAHGJa5gPSrjkYPb5baaRg4,
77
78
  bizyengine/misc/segment_anything.py,sha256=wNKYwlYPMszfwj23524geFZJjZaG4eye65SGaUnh77I,8941
78
79
  bizyengine/misc/supernode.py,sha256=STN9gaxfTSErH8OiHeZa47d8z-G9S0I7fXuJvHQOBFM,4532
79
80
  bizyengine/misc/utils.py,sha256=deQjBgLAkxIr-NaOMm77TcgBT3ExAp0MFm5ehUJ3CGs,6829
80
- bizyengine-1.2.33.dist-info/METADATA,sha256=wRBpNpUIpqJs8bk0BWy-FoxDUdZiH42YVmQElIQgAWg,675
81
- bizyengine-1.2.33.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
82
- bizyengine-1.2.33.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
83
- bizyengine-1.2.33.dist-info/RECORD,,
81
+ bizyengine-1.2.34.dist-info/METADATA,sha256=jnpLtDD4uqJ5B-INToxn3AWHqqejc9etvfiBS0_5StM,675
82
+ bizyengine-1.2.34.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
83
+ bizyengine-1.2.34.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
84
+ bizyengine-1.2.34.dist-info/RECORD,,