Update nodes.py

2024-12-04 03:14:07 +02:00
parent fbe38e011f
commit 0dd51cd53e
1 changed files with 1 additions and 1 deletions
@@ -647,7 +647,7 @@ class HyVideoSampler:
        #with autocast_context:
        if model["block_swap_args"] is not None:
            model["pipe"].transformer.to(device)
-            model["pipe"].transformer.block_swap(20, 0)
+            model["pipe"].transformer.block_swap(model["block_swap_args"]["double_blocks_to_swap"] , model["block_swap_args"]["single_blocks_to_swap"])
        elif model["manual_offloading"]:
            model["pipe"].transformer.to(device)