User specified an unsupported autocast device_type 'cuda:0'

[langbowang]

ComfyUI Error Report

Error Details

• Node Type: Joy_caption
• Exception Type: RuntimeError

• Exception Message: User specified an unsupported autocast device_type 'cuda:0'

  Stack Trace

  
  File "C:\Users\lbw\AIGC\ComfyUI\execution.py", line 317, in execute
  output_data, output_ui, has_subgraph = get_output_data(obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb)
                                         ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  
  File "C:\Users\lbw\AIGC\ComfyUI\execution.py", line 192, in get_output_data
  return_values = _map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb)
                  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  
  File "C:\Users\lbw\AIGC\ComfyUI\execution.py", line 169, in _map_node_over_list
  process_inputs(input_dict, i)
  
  File "C:\Users\lbw\AIGC\ComfyUI\execution.py", line 158, in process_inputs
  results.append(getattr(obj, func)(**inputs))
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  
  File "C:\Users\lbw\AIGC\ComfyUI\custom_nodes\ComfyUI_HF_Servelress_Inference\nodes\Joy_Caption.py", line 171, in gen
  with torch.amp.autocast_mode.autocast(DEVICE, enabled=True):
       ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  
  File "C:\Users\lbw\.conda\envs\F-002\Lib\site-packages\torch\amp\autocast_mode.py", line 241, in __init__
  raise RuntimeError(

[nosdregamon]

Got the same error, even when switching to cpu

File ".../custom_nodes/ComfyUI_HF_Servelress_Inference/nodes/Joy_Caption.py", line 171, in gen with torch.amp.autocast_mode.autocast(DEVICE, enabled=True): File ".../ComfyUI_windows_portable/python_embeded/Lib/site-packages/torch/amp/autocast_mode.py", line 241, in init raise RuntimeError( RuntimeError: User specified an unsupported autocast device_type 'cpu'

Perhaps a "autocast_mode.py" problem?

[specblades]

same here

[AlexBenjarmin]

No solution？

[kesyn]

You can fix it by modifying the Joy_caption_node.py file. Specifically, change the line:

with torch.amp.autocast_mode.autocast(str(DEVICE), enabled=True):

to

with torch.amp.autocast_mode.autocast(device_type=DEVICE.type, enabled=True):

This change ensures that autocast receives the correct device type ('cuda' or 'cpu') instead of the specific device name ('cuda:0').

[Positliver]

with torch.amp.autocast_mode.autocast(DEVICE, enabled=True): to with torch.cuda.amp.autocast(enabled=True):

[AlexBenjarmin]

with torch.amp.autocast_mode.autocast(DEVICE, enabled=True): to with torch.cuda.amp.autocast(enabled=True):

which file? Joy_caption_node.py?

[Positliver]

with torch.amp.autocast_mode.autocast(DEVICE, enabled=True): to with torch.cuda.amp.autocast(enabled=True):

which file? Joy_caption_node.py?

yes, Joy_caption_node.py

[Gonng]

You can fix it by modifying the Joy_caption_node.py file. Specifically, change the line:

with torch.amp.autocast_mode.autocast(str(DEVICE), enabled=True):

to

with torch.amp.autocast_mode.autocast(device_type=DEVICE.type, enabled=True):

This change ensures that autocast receives the correct device type ('cuda' or 'cpu') instead of the specific device name ('cuda:0').

thank you,it works for me!改完这一行代码，保存后，就正常运行了

[specblades]

Doesnt work for me. Nor with torch.cuda.amp.autocast(enabled=True): nor with torch.amp.autocast_mode.autocast(device_type=DEVICE.type, enabled=True):

[eoris]

replacing DEVICE = get_torch_device() with DEVICE = torch.device("cuda") also can help