@@ -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