From 48dae15e4a38f965eb0227c832b5d5e4f9b72a0c Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 24 Mar 2025 11:41:33 +0200
Subject: [PATCH] Add turbo paint model to loader

---
 nodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index e88dd9b..7dfe770 100644
--- a/nodes.py
+++ b/nodes.py
@@ -315,7 +315,7 @@ class DownloadAndLoadHy3DPaintModel:
     def INPUT_TYPES(s):
         return {
             "required": {
-                "model": (["hunyuan3d-paint-v2-0"],),
+                "model": (["hunyuan3d-paint-v2-0", "hunyuan3d-paint-v2-0-turbo"],),
             },
             "optional": {
                 "compile_args": ("HY3DCOMPILEARGS", {"tooltip": "torch.compile settings, when connected to the model loader, torch.compile of the selected models is attempted. Requires Triton and torch 2.5.0 is recommended"}),
@@ -340,7 +340,7 @@ class DownloadAndLoadHy3DPaintModel:
             snapshot_download(
                 repo_id="tencent/Hunyuan3D-2",
                 allow_patterns=[f"*{model}*"],
-                ignore_patterns=["*diffusion_pytorch_model.bin"],
+                ignore_patterns=["*unet/diffusion_pytorch_model.bin", "*image_encoder*"],
                 local_dir=download_path,
                 local_dir_use_symlinks=False,
             )