nodes/kijai-ComfyUI-segment-anything-2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix single image segment offload

Browse Source
This commit is contained in:
kijai
2025-09-28 11:35:08 +03:00
parent ebdb986959
commit 56abdb5a1e
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+3 -2
View File
@@ -411,8 +411,9 @@ class Sam2Segmentation:
model.to(offload_device) model.to(offload_device)
except: except:
model.model.to(offload_device) model.model.to(offload_device)
model.reset_state(self.inference_state) if hasattr(self, 'inference_state') and self.inference_state is not None and hasattr(model, "reset_state"):
self.inference_state = None model.reset_state(self.inference_state)
self.inference_state = None
mm.soft_empty_cache() mm.soft_empty_cache()
out_list = [] out_list = []