Skip to content

Commit 0618b7c

Browse files
usophiaanusophiaan
authored
Merge pull request #1 from mfzzf/main
update version to 1.0.4
2 parents d0c9b9f + bf07821 commit 0618b7c

19 files changed

+933
-98
lines changed

‎.gitignore‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
.DS_Store
2+
__pycache__
3+
py/__pycache__
4+
py/modelverse_api/__pycache__
5+
py/modelverse_api/requests/__pycache__
6+
/venv
7+
.idea/

‎__init__.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,8 @@ def serialize(obj):
3939
serialized_CLASS_MAPPINGS = {k: serialize(v) for k, v in imported_module.NODE_CLASS_MAPPINGS.items()}
4040
serialized_DISPLAY_NAME_MAPPINGS = {k: serialize(v) for k, v in imported_module.NODE_DISPLAY_NAME_MAPPINGS.items()}
4141
all_nodes[file]={"NODE_CLASS_MAPPINGS": serialized_CLASS_MAPPINGS, "NODE_DISPLAY_NAME_MAPPINGS": serialized_DISPLAY_NAME_MAPPINGS}
42-
except:
42+
except Exception as e:
43+
print(f"Failed to import {file}: {e}")
4344
pass
4445

4546

‎py/flux_dev.py‎

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from .modelverse_api.client import ModelverseClient
33
from .modelverse_api.requests.flux_dev import FluxDev
44
import torch
5-
import asyncio
5+
from comfy.comfy_types.node_typing import IO
66

77

88
class FluxDevNode:
@@ -17,63 +17,63 @@ def INPUT_TYPES(s):
1717
return {
1818
"required": {
1919
"client": ("MODELVERSE_API_CLIENT",),
20-
"prompt": ("STRING", {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
21-
"width": ("INT", {
20+
"prompt": (IO.STRING, {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
21+
"width": (IO.INT, {
2222
"default": 1024,
2323
"min": 512,
2424
"max": 1536,
2525
"step": 8,
2626
"display": "number",
2727
"tooltip": "Image width (512 to 1536)"
2828
}),
29-
"height": ("INT", {
29+
"height": (IO.INT, {
3030
"default": 1024,
3131
"min": 512,
3232
"max": 1536,
3333
"step": 8,
3434
"display": "number",
3535
"tooltip": "Image height (512 to 1536)"
3636
}),
37-
"strength": ("FLOAT", {
37+
"strength": (IO.FLOAT, {
3838
"default": 0.8,
3939
"min": 0.01,
4040
"max": 1.0,
4141
"step": 0.01,
4242
"display": "number",
4343
"tooltip": "Strength of the image-to-image transformation (0.01 to 1.0)"
4444
}),
45-
"seed": ("INT", {
45+
"seed": (IO.INT, {
4646
"default": -1,
4747
"min": -1,
4848
"max": 0xffffffffffffffff,
4949
"control_after_generate": True,
5050
"tooltip": "Random seed for reproducible results. -1 for random seed"
5151
}),
52-
"num_images": ("INT", {
52+
"num_images": (IO.INT, {
5353
"default": 1,
5454
"min": 1,
5555
"max": 4,
5656
"step": 1,
5757
"display": "number",
5858
"tooltip": "Number of images to generate in a single request (1 to 4)"
5959
}),
60-
"num_requests": ("INT", {
60+
"num_requests": (IO.INT, {
6161
"default": 1,
6262
"min": 1,
6363
"max": 10,
6464
"step": 1,
6565
"display": "number",
6666
"tooltip": "Number of request to make (1 to 10)"
6767
}),
68-
"num_inference_steps": ("INT", {
68+
"num_inference_steps": (IO.INT, {
6969
"default": 28,
7070
"min": 1,
7171
"max": 50,
7272
"step": 1,
7373
"display": "number",
7474
"tooltip": "Number of inference steps (1 to 50)"
7575
}),
76-
"guidance_scale": ("FLOAT", {
76+
"guidance_scale": (IO.FLOAT, {
7777
"default": 3.5,
7878
"min": 0.0,
7979
"max": 10.0,
@@ -83,7 +83,7 @@ def INPUT_TYPES(s):
8383
}),
8484
},
8585
"optional": {
86-
"image": ("IMAGE", {
86+
"image": (IO.IMAGE, {
8787
"tooltip": "The image for reference.",
8888
"forceInput": False,
8989
"default": None

‎py/flux_kontext_max.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from .modelverse_api.client import ModelverseClient
33
from .modelverse_api.requests.flux_kontext_max import FluxKontextMax, FluxKontextMaxMulti
44
import torch
5-
import asyncio
5+
from comfy.comfy_types.node_typing import IO
66

77

88
class FluxKontextMaxNode:
@@ -21,28 +21,28 @@ def INPUT_TYPES(s):
2121
return {
2222
"required": {
2323
"client": ("MODELVERSE_API_CLIENT",),
24-
"prompt": ("STRING", {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
25-
"images": ("IMAGE,IMAGE_LIST", {
24+
"prompt": (IO.STRING, {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
25+
"images": ("IMAGE_LIST", {
2626
"tooltip": "The image(s) to edit from. If not included, use text-to-image mode.",
2727
"forceInput": False,
2828
"default": None
2929
}),
30-
"num_requests": ("INT", {
30+
"num_requests": (IO.INT, {
3131
"default": 1,
3232
"min": 1,
3333
"max": 10,
3434
"step": 1,
3535
"display": "number",
3636
"tooltip": "Number of request to make (1 to 10)"
3737
}),
38-
"seed": ("INT", {
38+
"seed": (IO.INT, {
3939
"default": -1,
4040
"min": -1,
4141
"max": 0xffffffffffffffff,
4242
"control_after_generate": True,
4343
"tooltip": "Random seed for reproducible results. -1 for random seed"
4444
}),
45-
"guidance_scale": ("FLOAT", {
45+
"guidance_scale": (IO.FLOAT, {
4646
"default": 2.5,
4747
"min": 0.0,
4848
"max": 10.0,

‎py/flux_kontext_max_t2i.py‎

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@
22
from .modelverse_api.client import ModelverseClient
33
from .modelverse_api.requests.flux_kontext_max import FluxKontextMaxT2I
44
import torch
5-
import asyncio
6-
5+
from comfy.comfy_types.node_typing import IO
76

87
class FluxKontextMaxT2INode:
98
"""
@@ -18,35 +17,35 @@ def INPUT_TYPES(s):
1817
return {
1918
"required": {
2019
"client": ("MODELVERSE_API_CLIENT",),
21-
"prompt": ("STRING", {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
20+
"prompt": (IO.STRING, {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
2221
"aspect_ratio": (["21:9", "16:9", "16:10", "4:3", "1:1", "3:4", "10:16", "9:16", "9:21"], {
2322
"default": "1:1",
2423
"tooltip": "The aspect ratio of the output image, ranging from \"21:9\" to \"9:21\", default is \"1:1\""
2524
}),
26-
"num_images": ("INT", {
25+
"num_images": (IO.INT, {
2726
"default": 1,
2827
"min": 1,
2928
"max": 4,
3029
"step": 1,
3130
"display": "number",
3231
"tooltip": "Number of images to generate in a single request (1 to 4)"
3332
}),
34-
"num_requests": ("INT", {
33+
"num_requests": (IO.INT, {
3534
"default": 1,
3635
"min": 1,
3736
"max": 10,
3837
"step": 1,
3938
"display": "number",
4039
"tooltip": "Number of request to make (1 to 10)"
4140
}),
42-
"seed": ("INT", {
41+
"seed": (IO.INT, {
4342
"default": -1,
4443
"min": -1,
4544
"max": 0xffffffffffffffff,
4645
"control_after_generate": True,
4746
"tooltip": "Random seed for reproducible results. -1 for random seed"
4847
}),
49-
"guidance_scale": ("FLOAT", {
48+
"guidance_scale": (IO.FLOAT, {
5049
"default": 2.5,
5150
"min": 0.0,
5251
"max": 10.0,

‎py/flux_kontext_pro.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from .modelverse_api.client import ModelverseClient
33
from .modelverse_api.requests.flux_kontext_pro import FluxKontextPro, FluxKontextProMulti
44
import torch
5-
import asyncio
5+
from comfy.comfy_types.node_typing import IO
66

77

88
class FluxKontextProNode:
@@ -21,28 +21,28 @@ def INPUT_TYPES(s):
2121
return {
2222
"required": {
2323
"client": ("MODELVERSE_API_CLIENT",),
24-
"prompt": ("STRING", {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
25-
"images": ("IMAGE,IMAGE_LIST", {
24+
"prompt": (IO.STRING, {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
25+
"images": ("IMAGE_LIST", {
2626
"tooltip": "The image(s) to edit from. If not included, use text-to-image mode.",
2727
"forceInput": False,
2828
"default": None
2929
}),
30-
"num_requests": ("INT", {
30+
"num_requests": (IO.INT, {
3131
"default": 1,
3232
"min": 1,
3333
"max": 10,
3434
"step": 1,
3535
"display": "number",
3636
"tooltip": "Number of request to make (1 to 10)"
3737
}),
38-
"seed": ("INT", {
38+
"seed": (IO.INT, {
3939
"default": -1,
4040
"min": -1,
4141
"max": 0xffffffffffffffff,
4242
"control_after_generate": True,
4343
"tooltip": "Random seed for reproducible results. -1 for random seed"
4444
}),
45-
"guidance_scale": ("FLOAT", {
45+
"guidance_scale": (IO.FLOAT, {
4646
"default": 2.5,
4747
"min": 0.0,
4848
"max": 10.0,

‎py/flux_kontext_pro_t2i.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from .modelverse_api.client import ModelverseClient
33
from .modelverse_api.requests.flux_kontext_pro import FluxKontextProT2I
44
import torch
5-
import asyncio
5+
from comfy.comfy_types.node_typing import IO
66

77

88
class FluxKontextProT2INode:
@@ -18,35 +18,35 @@ def INPUT_TYPES(s):
1818
return {
1919
"required": {
2020
"client": ("MODELVERSE_API_CLIENT",),
21-
"prompt": ("STRING", {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
21+
"prompt": (IO.STRING, {"multiline": True, "default": "", "tooltip": "Text description of the image to generate"}),
2222
"aspect_ratio": (["21:9", "16:9", "16:10", "4:3", "1:1", "3:4", "10:16", "9:16", "9:21"], {
2323
"default": "1:1",
2424
"tooltip": "The aspect ratio of the output image, ranging from \"21:9\" to \"9:21\", default is \"1:1\""
2525
}),
26-
"num_images": ("INT", {
26+
"num_images": (IO.INT, {
2727
"default": 1,
2828
"min": 1,
2929
"max": 4,
3030
"step": 1,
3131
"display": "number",
3232
"tooltip": "Number of images to generate in a single request (1 to 4)"
3333
}),
34-
"num_requests": ("INT", {
34+
"num_requests": (IO.INT, {
3535
"default": 1,
3636
"min": 1,
3737
"max": 10,
3838
"step": 1,
3939
"display": "number",
4040
"tooltip": "Number of request to make (1 to 10)"
4141
}),
42-
"seed": ("INT", {
42+
"seed": (IO.INT, {
4343
"default": -1,
4444
"min": -1,
4545
"max": 0xffffffffffffffff,
4646
"control_after_generate": True,
4747
"tooltip": "Random seed for reproducible results. -1 for random seed"
4848
}),
49-
"guidance_scale": ("FLOAT", {
49+
"guidance_scale": (IO.FLOAT, {
5050
"default": 2.5,
5151
"min": 0.0,
5252
"max": 10.0,

0 commit comments

Comments
 (0)