Updated IP2V with Device Selection

kijai · Dec 19, 2024 · bc55ee9 · bc55ee9
1 parent 7341319
commit bc55ee9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nodes.py b/nodes.py
@@ -855,6 +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"}),
             "image_token_selection_expr": ("STRING", {"default": "::4", "multiline": False} ),
             },
             "optional": {