Update from Pull Request #122 review

2025-10-10 15:11:25 -05:00
parent e4c2006c44
commit 830342453c
1 changed files with 1 additions and 1 deletions
@@ -857,7 +857,7 @@ class WanVideoUni3C_ControlnetLoader:
    FUNCTION = "loadmodel"
    CATEGORY = "multigpu/WanVideoWrapper"

-    def loadmodel(self, model, base_precision, load_device, quantization, attention_mode, compile_args=None):
+    def loadmodel(self, model, base_precision, load_device, device, quantization, attention_mode, compile_args=None):
        from . import set_current_device

        set_current_device(device)