ComfyUI-SAM2
ComfyUI-SAM2 copied to clipboard
'ImageEncoder' object has no attribute 'img_size'
If I use a prompt like horse, an error will be reported, but if I use something like cat that does not exist on the screen, a whole black picture will be output normally.
Trace:
!!! Exception during processing!!! 'ImageEncoder' object has no attribute 'img_size'
Traceback (most recent call last):
File "C:\Users\User\Desktop\ComfyUI\execution.py", line 152, in recursive_execute
output_data, output_ui = get_output_data(obj, input_data_all)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\User\Desktop\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 "C:\Users\User\Desktop\ComfyUI\execution.py", line 75, in map_node_over_list
results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\User\Desktop\ComfyUI\custom_nodes\comfyui_segment_anything\node.py", line 325, in main
(images, masks) = sam_segment(
^^^^^^^^^^^^
File "C:\Users\User\Desktop\ComfyUI\custom_nodes\comfyui_segment_anything\node.py", line 240, in sam_segment
predictor = SamPredictorHQ(sam_model, sam_is_hq)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\User\Desktop\ComfyUI\custom_nodes\comfyui_segment_anything\sam_hq\predictor.py", line 22, in __init__
super().__init__(sam_model=sam_model)
File "C:\Users\User\AppData\Local\Programs\Python\Python312\Lib\site-packages\segment_anything\predictor.py", line 31, in __init__
self.transform = ResizeLongestSide(sam_model.image_encoder.img_size)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\User\AppData\Local\Programs\Python\Python312\Lib\site-packages\torch\nn\modules\module.py", line 1729, in __getattr__
raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'")
AttributeError: 'ImageEncoder' object has no attribute 'img_size'