Add turbo paint model to loader

This commit is contained in:
kijai 2025-03-24 11:41:33 +02:00
parent 921b0d78a9
commit 48dae15e4a

View File

@ -315,7 +315,7 @@ class DownloadAndLoadHy3DPaintModel:
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
"required": { "required": {
"model": (["hunyuan3d-paint-v2-0"],), "model": (["hunyuan3d-paint-v2-0", "hunyuan3d-paint-v2-0-turbo"],),
}, },
"optional": { "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"}), "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( snapshot_download(
repo_id="tencent/Hunyuan3D-2", repo_id="tencent/Hunyuan3D-2",
allow_patterns=[f"*{model}*"], 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=download_path,
local_dir_use_symlinks=False, local_dir_use_symlinks=False,
) )