xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-16 08:14:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update mask_nodes.py

Browse Source
This commit is contained in:
Mokan Alexander 2024-05-10 03:39:06 +03:00 committed by GitHub
parent 3260038935
commit 030b6a20b5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nodes/mask_nodes.py
View File

@ -91,6 +91,9 @@ Segments an image or batch of images using CLIPSeg.
if binary_mask: if binary_mask:
results = results.round() results = results.round()
del outputs, tensor, tensor_thresholded, tensor_normalized, resized_tensor, images
torch.cuda.empty_cache()
return results, return results,
class CreateTextMask: class CreateTextMask: