neverbiasu / ComfyUI-SAM2

A ComfyUI extension for Segment-Anything 2
72 stars 13 forks source link

Allocation on device - Unable to download SAM Model Loader And GroundingDINO #14

Open TheMindExpansionNetwork opened 2 months ago

TheMindExpansionNetwork commented 2 months ago

Hello you all I am trying to download and get this error

Error occurred when executing SAMModelLoader (segment anything):

Allocation on device

File "/teamspace/studios/this_studio/ComfyUI/execution.py", line 152, in recursive_execute output_data, output_ui = get_output_data(obj, input_data_all) File "/teamspace/studios/this_studio/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 "/teamspace/studios/this_studio/ComfyUI/execution.py", line 75, in map_node_over_list results.append(getattr(obj, func)(**slice_dict(input_data_all, i))) File "/teamspace/studios/this_studio/ComfyUI/custom_nodes/comfyui_segment_anything/node.py", line 269, in main sam_model = load_sam_model(model_name) File "/teamspace/studios/this_studio/ComfyUI/custom_nodes/comfyui_segment_anything/node.py", line 88, in load_sam_model sam.to(device=sam_device) File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1174, in to return self._apply(convert) File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, in _apply module._apply(fn) File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, in _apply module._apply(fn) File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, in _apply module._apply(fn) [Previous line repeated 2 more times] File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/torch/nn/modules/module.py", line 805, in _apply param_applied = fn(param) File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1160, in convert return t.to( image

image

I am using Lighting AI pod not sure that this the issue but everything works but this

Does anyone know what the issue could be?

neverbiasu commented 2 months ago

u r using comfyui-segement-anything, u need to install ComfyUI-SAM2 to use this node.