xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2026-05-13 14:06:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2025-01-31 17:24:05 +02:00
parent dbd5f09c11
commit 6fb6cba50b
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
nodes.py
View File

@ -327,6 +327,7 @@ class DownloadAndLoadHy3DPaintModel:
) )
if compile_args is not None: if compile_args is not None:
pipeline.to(device)
torch._dynamo.config.cache_size_limit = compile_args["dynamo_cache_size_limit"] torch._dynamo.config.cache_size_limit = compile_args["dynamo_cache_size_limit"]
if compile_args["compile_transformer"]: if compile_args["compile_transformer"]:
pipeline.unet = torch.compile(pipeline.unet) pipeline.unet = torch.compile(pipeline.unet)