offload and support lora

.gitignore

@@ -159,3 +159,4 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 *.zip
+.tracking

nodes/nodes.py

@@ -147,7 +147,8 @@ class WanImageToVideo_F2:
     CATEGORY = "Flow2/Wan 2.1"
 
     def encode(self, positive, negative, vae, width, height, length, start_image=None, end_image=None, clip_vision_output=None):
-        device = mm.intermediate_device()
+        offload_device = mm.unet_offload_device()
+        device = offload_device
 
         valid_end_image = end_image is not None
 

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "ComfyUI-WanStartEndFramesNative"
 description = "Start and end frames video generation node that supports native ComfyUI."
-version = "1.0.1"
+version = "1.0.2"
 license = {file = "LICENSE"}
 dependencies = []