NameError: name 'sam3_model' is not defined in AILab_SAM3Segment.py

[lovelybbq]

Hello. I've tried running "SAM3 Segmentation (RMBG)" node and this is the error I got. I updated nodes recently, so I believe some update broke it, because some time ago it worked fine.

Traceback (most recent call last):
  File "D:\ComfyUI_windows_portable\ComfyUI\execution.py", line 518, in execute
    output_data, output_ui, has_subgraph, has_pending_tasks = await get_output_data(prompt_id, unique_id, obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
                                                              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\ComfyUI_windows_portable\ComfyUI\execution.py", line 329, in get_output_data
    return_values = await _async_map_node_over_list(prompt_id, unique_id, obj, input_data_all, obj.FUNCTION, allow_interrupt=True, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-lora-manager\py\metadata_collector\metadata_hook.py", line 165, in async_map_node_over_list_with_metadata
    results = await original_map_node_over_list(
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\ComfyUI_windows_portable\ComfyUI\execution.py", line 303, in _async_map_node_over_list
    await process_inputs(input_dict, i)
  File "D:\ComfyUI_windows_portable\ComfyUI\execution.py", line 291, in process_inputs
    result = f(**inputs)
             ^^^^^^^^^^^
  File "D:\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-rmbg\py\AILab_SAM3Segment.py", line 307, in segment
    cache_key = (sam3_model, device_str)
                 ^^^^^^^^^^
NameError: name 'sam3_model' is not defined

File: custom_nodes/comfyui-rmbg/py/AILab_SAM3Segment.py
Line: 307
Code: cache_key = (sam3_model, device_str)

[masterkamy568-coder]

将AILab_SAM3Segment.py的307行代码改成：cache_key = ("sam3", device_str)，保持与load_processor中使用的缓存键一致应该就没问题了

[lovelybbq]

It worked. Thank you

[1038lab]

Thanks for the report.
This was caused by an incorrect cache key when unload_model=True.
The unload logic referenced an undefined variable (sam3_model).

The fix is to use the same cache key as _load_processor():

cache_key = ("sam3", device_str) 

This has been confirmed to resolve the issue. Thanks for confirming 👍