Skip to content

Commit d113d1c

Browse files
authored
feat(api-nodes-Tencent3D): allow smaller possible face_count; add uv_image output (Comfy-Org#13207)
Signed-off-by: bigcat88 <bigcat88@icloud.com>
1 parent a500f1e commit d113d1c

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

comfy_api_nodes/nodes_hunyuan3d.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def define_schema(cls):
132132
tooltip="The LowPoly option is unavailable for the `3.1` model.",
133133
),
134134
IO.String.Input("prompt", multiline=True, default="", tooltip="Supports up to 1024 characters."),
135-
IO.Int.Input("face_count", default=500000, min=40000, max=1500000),
135+
IO.Int.Input("face_count", default=500000, min=3000, max=1500000),
136136
IO.DynamicCombo.Input(
137137
"generate_type",
138138
options=[
@@ -251,7 +251,7 @@ def define_schema(cls):
251251
IO.Image.Input("image_left", optional=True),
252252
IO.Image.Input("image_right", optional=True),
253253
IO.Image.Input("image_back", optional=True),
254-
IO.Int.Input("face_count", default=500000, min=40000, max=1500000),
254+
IO.Int.Input("face_count", default=500000, min=3000, max=1500000),
255255
IO.DynamicCombo.Input(
256256
"generate_type",
257257
options=[
@@ -422,6 +422,7 @@ def define_schema(cls):
422422
outputs=[
423423
IO.File3DOBJ.Output(display_name="OBJ"),
424424
IO.File3DFBX.Output(display_name="FBX"),
425+
IO.Image.Output(display_name="uv_image"),
425426
],
426427
hidden=[
427428
IO.Hidden.auth_token_comfy_org,
@@ -468,9 +469,16 @@ async def execute(
468469
response_model=To3DProTaskResultResponse,
469470
status_extractor=lambda r: r.Status,
470471
)
472+
uv_image_file = get_file_from_response(result.ResultFile3Ds, "uv_image", raise_if_not_found=False)
473+
uv_image = (
474+
await download_url_to_image_tensor(uv_image_file.Url)
475+
if uv_image_file is not None
476+
else torch.zeros(1, 1, 1, 3)
477+
)
471478
return IO.NodeOutput(
472479
await download_url_to_file_3d(get_file_from_response(result.ResultFile3Ds, "obj").Url, "obj"),
473480
await download_url_to_file_3d(get_file_from_response(result.ResultFile3Ds, "fbx").Url, "fbx"),
481+
uv_image,
474482
)
475483

476484

0 commit comments

Comments
 (0)