cannot select an axis to squeeze out which has size not equal to one
- 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

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
Assessment
This issue has not been assessed yet.