2kpr / ComfyUI-PMRF

GNU Affero General Public License v3.0
55 stars 2 forks source link

ModuleNotFoundError: natten is required for neighborhood attention #9

Open Song367 opened 1 week ago

Song367 commented 1 week ago

!!! Exception during processing !!! natten is required for neighborhood attention Traceback (most recent call last): File "E:\workplace\ComfyUI\execution.py", line 323, 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 "E:\workplace\ComfyUI\execution.py", line 198, 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 "E:\workplace\ComfyUI\execution.py", line 169, in _map_node_over_list process_inputs(input_dict, i) File "E:\workplace\ComfyUI\execution.py", line 158, in process_inputs results.append(getattr(obj, func)(inputs)) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\nodes.py", line 190, in pmrf return inference(images, scale, num_steps, seed, interpolation) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\utils_contextlib.py", line 116, in decorate_context return func(*args, *kwargs) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\nodes.py", line 153, in inference cropped_face, restored_faces, restored_img = enhance_face( File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\utils_contextlib.py", line 116, in decorate_context return func(args, kwargs) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\nodes.py", line 84, in enhance_face output = generate_reconstructions( File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\nodes.py", line 60, in generate_reconstructions v_t_next = pmrf_model(x_t=x_t_next, t=t_one num_t, y=y).to(x_t_next.dtype) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1553, in _wrapped_call_impl return self._call_impl(args, kwargs) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1562, in _call_impl return forward_call(*args, *kwargs) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\lightning_models\mmse_rectified_flow.py", line 135, in forward return self.forward_flow(x_t, t, y) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\lightning_models\mmse_rectified_flow.py", line 131, in forward_flow return self.model(x_t, t) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1553, in _wrapped_call_impl return self._call_impl(args, kwargs) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1562, in _call_impl return forward_call(*args, kwargs) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\arch\hourglass\image_transformer_v2.py", line 749, in forward x = down_level(x, pos, cond) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1553, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1562, in _call_impl return forward_call(args, kwargs) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\arch\hourglass\image_transformer_v2.py", line 542, in forward x = layer(x, *args, kwargs) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1553, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1562, in _call_impl return forward_call(args, kwargs) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\arch\hourglass\image_transformer_v2.py", line 511, in forward x = checkpoint(self.self_attn, x, pos, cond) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\arch\hourglass\image_transformer_v2.py", line 50, in checkpoint return function(*args, kwargs) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1553, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "E:\workplace\ComfyUI\venvs\lib\site-packages\torch\nn\modules\module.py", line 1562, in _call_impl return forward_call(args, kwargs) File "E:\workplace\ComfyUI\custom_nodes\ComfyUI-PMRF\arch\hourglass\image_transformer_v2.py", line 416, in forward raise ModuleNotFoundError("natten is required for neighborhood attention") ModuleNotFoundError: natten is required for neighborhood attention

Prompt executed in 2.68 seconds