Update nodes.py

2026-03-16 16:27:12 +08:00 · 2025-01-31 17:24:05 +02:00 · 2025-01-31 17:24:05 +02:00 · 6fb6cba50b
commit 6fb6cba50b
parent dbd5f09c11
1 changed files with 1 additions and 0 deletions
--- a/nodes.py
+++ b/nodes.py
@ -327,6 +327,7 @@ class DownloadAndLoadHy3DPaintModel:
            )
        
        if compile_args is not None:
+            pipeline.to(device)
            torch._dynamo.config.cache_size_limit = compile_args["dynamo_cache_size_limit"]
            if compile_args["compile_transformer"]:
                pipeline.unet = torch.compile(pipeline.unet)