Added variant support.

Author: niknah

hunyuan_3d_node.py

@@ -20,7 +20,7 @@ class Hunyuan3DImageTo3D:
     def INPUT_TYPES(s):
         models = [
             'tencent/Hunyuan3D-2/hunyuan3d-dit-v2-0',
-            'tencent/Hunyuan3D-2/hunyuan3d-dit-v2-0-fast',
+            'tencent/Hunyuan3D-2/hunyuan3d-dit-v2-0-fast[variant=fp16]',
             ]
         return {
             "required": {
@@ -37,7 +37,7 @@ def INPUT_TYPES(s):
                     "tooltip": "Paint needs a lot more VRAM",
                 }),
                 "model": (models, {
-                    "tooltip": "huggingface id of model.  ie. author/name/subfolder",
+                    "tooltip": "huggingface id of model(author/name/subfolder)",  # noqa: E501
                 }),
             }
         }
@@ -140,6 +140,12 @@ def process(
 #        if 'HY3DGEN_MODELS' not in os.environ:
 #            os.environ['HY3DGEN_MODELS'] = checkpoint_dir
 
+        variant = None
+        variant_m = re.match(r"^(.*)\[variant=([^\]]+)\]$", model)
+        if variant_m is not None:
+            model = variant_m.group(1)
+            variant = variant_m.group(2)
+
         subfolder = None
         subfolder_m = re.match(r"^([^/]+/[^/]+)/(.+)$", model)
         if subfolder_m is not None:
@@ -167,6 +173,7 @@ def process(
                 pipeline = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(
                     model,
                     subfolder=subfolder,
+                    variant=variant
                 )
 
                 output_3d_file = Hunyuan3DImageTo3D.get_spare_filename(
@@ -186,7 +193,6 @@ def process(
                     from hy3dgen.texgen import Hunyuan3DPaintPipeline
                     pipeline = Hunyuan3DPaintPipeline.from_pretrained(
                         model,
-                        subfolder=subfolder,
                     )
                     mesh = pipeline(mesh, image=input_image_file)
 

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-hunyuan-3d-2"
 description = "Image to 3D using Hunyuan-3D-2"
-version = "0.9.2"
+version = "0.9.3"
 license = {text = "MIT License"}
 
 [project.urls]