IP2V

Fixed IP2V
kijai · Dec 19, 2024 · 561c058 · 561c058
1 parent bc55ee9
commit 561c058
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nodes.py b/nodes.py
@@ -855,7 +855,7 @@ def INPUT_TYPES(s):
         return {"required": {
             "text_encoders": ("HYVIDTEXTENCODER",),
             "prompt": ("STRING", {"default": "", "multiline": True} ),
-            "load_device": (["main_device", "offload_device"], {"default": "main_device"}),
+            "device": ([f"cuda:{i}" for i in range(torch.cuda.device_count())],),
             "image_token_selection_expr": ("STRING", {"default": "::4", "multiline": False} ),
             },
             "optional": {