Comfy-Org / Comfy-Org/ComfyUI

cannot select an axis to squeeze out which has size not equal to one

Open
#1,665 4 comments 0 reactions 0 assignees View on GitHub
User Support
Dominant language
Python
Stars
133k
Forks
15.7k
Avg merge
1d 6h
Merged PRs (30d)
155

Description

I upgraded to version: 9bfec2bdbf0b0d778087a9b32f79e57e2d15b913
The following error occurs:

Loads SAM model: /mnt/workspace/ComfyUI/models/sams/VIT-B/sam_vit_b_01ec64.pth (device:AUTO)

0: 640x448 1 person, 89.3ms
Speed: 2.9ms preprocess, 89.3ms inference, 2.6ms postprocess per image at shape (1, 3, 640, 448)
semd to cuda:0
Loads SAM model: /mnt/workspace/ComfyUI/models/sams/VIT-B/sam_vit_b_01ec64.pth (device:AUTO)
semd to cuda:0
# of Detected SEGS: 1
!!! Exception during processing !!!
Traceback (most recent call last):
File "/mnt/workspace/ComfyUI/execution.py", line 152, in recursive_execute
output_data, output_ui = get_output_data(obj, input_data_all)
File "/mnt/workspace/ComfyUI/execution.py", line 82, in get_output_data
return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True)
File "/mnt/workspace/ComfyUI/execution.py", line 75, in map_node_over_list
results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
File "/mnt/workspace/ComfyUI/custom_nodes/ComfyUI-Impact-Pack/modules/impact/impact_pack.py", line 1918, in doit
result = core.mask_to_segs(mask, combined, crop_factor, bbox_fill, drop_size)
File "/mnt/workspace/ComfyUI/custom_nodes/ComfyUI-Impact-Pack/modules/impact/core.py", line 712, in mask_to_segs
mask = np.squeeze(mask, axis=1)
File "<__array_function__ internals>", line 200, in squeeze
File "/opt/conda/lib/python3.10/site-packages/numpy/core/fromnumeric.py", line 1571, in squeeze
return squeeze(axis=axis)
ValueError: cannot select an axis to squeeze out which has size not equal to one
![image](https://github.com/comfyanonymous/ComfyUI/assets/139383150/fae2dc52-b3e7-4fef-9892-5cc7e8c5c626)

When I fall back to: 77c124c5a17534e347bdebbc1ace807d61416147
then together is normal. We used 3 machines, Linux and Windows, although 9bfec2bdbf0b0d778087a9b32f79e57e2d15b913 error is different, but 77c124c5a17534e347bdebbc1ace807d61416147 are normal.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.