zho-zho-zho/comfyui-segmoe

Unofficial implementation of SegMoE for ComfyUI

comfyui-segmoe's Issues

无论使用何种 "segmoe"，都会出现同样的错误

Error occurred when executing SMoE_Generation_Zho:

SparseMoeBlock.forward() missing 1 required positional argument: 'scale'

File "E:\comfyui_install\execution.py", line 152, in recursive_execute
output_data, output_ui = get_output_data(obj, input_data_all)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\execution.py", line 82, in get_output_data
return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\execution.py", line 75, in map_node_over_list
results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\custom_nodes\ComfyUI-SegMoE\SegMoE.py", line 68, in smoe_generate_image
output = pipe(
^^^^^
File "E:\comfyui_install\custom_nodes\ComfyUI-SegMoE\segmoe\main.py", line 865, in call
return self.pipe(*args, **kwds)
^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\utils_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\diffusers\pipelines\stable_diffusion\pipeline_stable_diffusion.py", line 1002, in call
noise_pred = self.unet(
^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\diffusers\models\unet_2d_condition.py", line 1112, in forward
sample, res_samples = downsample_block(
^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\diffusers\models\unet_2d_blocks.py", line 1160, in forward
hidden_states = attn(
^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\diffusers\models\transformer_2d.py", line 392, in forward
hidden_states = block(
^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\diffusers\models\attention.py", line 329, in forward
attn_output = self.attn1(
^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\diffusers\models\attention_processor.py", line 527, in forward
return self.processor(
^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\diffusers\models\attention_processor.py", line 1239, in call
query = attn.to_q(hidden_states, *args)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\comfyui_install\venv\Lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^