Error occurred when executing KolorsTextEncode

[Sostay]

Error occurred when executing KolorsTextEncode:

Allocation on device

File "D:\ComfyUI\execution.py", line 151, in recursive_execute
output_data, output_ui = get_output_data(obj, input_data_all)
File "D:\ComfyUI\execution.py", line 81, in get_output_data
return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True)
File "D:\ComfyUI\execution.py", line 74, in map_node_over_list
results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
File "D:\ComfyUI\custom_nodes\ComfyUI-KwaiKolorsWrapper\nodes.py", line 248, in encode
text_encoder.to(device)
File "D:\ComfyUI\python\lib\site-packages\transformers\modeling_utils.py", line 2692, in to
return super().to(*args, **kwargs)
File "D:\ComfyUI\python\lib\site-packages\torch\nn\modules\module.py", line 1173, in to
return self._apply(convert)
File "D:\ComfyUI\python\lib\site-packages\torch\nn\modules\module.py", line 779, in _apply
module._apply(fn)
File "D:\ComfyUI\python\lib\site-packages\torch\nn\modules\module.py", line 804, in _apply
param_applied = fn(param)
File "D:\ComfyUI\python\lib\site-packages\torch\nn\modules\module.py", line 1159, in convert
return t.to(