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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-04-23 00:52:48 +03:00
parent 60cd6b555a
commit 48d5a18bd4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes.py
View File

@ -2462,7 +2462,7 @@ Segments an image or batch of images using CLIPSeg.
for image in images: for image in images:
image = (image* 255).type(torch.uint8) image = (image* 255).type(torch.uint8)
prompt = text prompt = text
input_prc = processor(text=prompt, images=image, padding="max_length", return_tensors="pt") input_prc = processor(text=prompt, images=image, return_tensors="pt")
# Move the processed input to the device # Move the processed input to the device
for key in input_prc: for key in input_prc:
input_prc[key] = input_prc[key].to(device) input_prc[key] = input_prc[key].to(device)