'NoneType' object is not subscriptable

[NeilWang079]

Prompt executed in 2078.53 seconds Sampling: 0%| | 0/50 [00:00<?, ?it/s] got prompt torch.Size([1, 3, 13, 480, 848]) images before encoding torch.Size([1, 3, 13, 480, 848]) encoder output torch.Size([1, 12, 3, 60, 106]) Sampling: 0%| | 0/50 [00:00<?, ?it/s]!!! Exception during processing !!! 'NoneType' object is not subscriptable Traceback (most recent call last): File "E:\comfyui\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:\comfyui\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:\comfyui\ComfyUI\execution.py", line 169, in _map_node_over_list process_inputs(input_dict, i) File "E:\comfyui\ComfyUI\execution.py", line 158, in process_inputs results.append(getattr(obj, func)(inputs)) File "E:\comfyui\comfyui\custom_nodes\ComfyUI-MochiEdit\nodes\sampler_custom_node.py", line 40, in process latents = run_sampler(model, latents, positive, negative, sigmas, cfg, sampler.sampler_function, add_noise, seed) File "E:\comfyui\comfyui\custom_nodes\ComfyUI-MochiEdit\sampling\sampler.py", line 38, in run_sampler z = sampler_fn(model_fn, z, sigmas, callback=callback_fn, extra_args=extra_args) File "E:\comfyui\python_embeded\lib\site-packages\torch\utils_contextlib.py", line 116, in decorate_context return func(*args, *kwargs) File "E:\comfyui\comfyui\custom_nodes\ComfyUI-MochiEdit\sampling\sampling_functions.py", line 37, in sample_forward unconditional_vector_field = -model(Y, sigmas[i]s_in, extra_args) File "E:\comfyui\comfyui\custom_nodes\ComfyUI-MochiEdit\utils\sampling_utils.py", line 20, in model_fn out_cond = model.dit(z, sigma, positive) File "E:\comfyui\python_embeded\lib\site-packages\torch\nn\modules\module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "E:\comfyui\python_embeded\lib\site-packages\torch\nn\modules\module.py", line 1747, in _call_impl return forward_call(args, kwargs) File "E:\comfyui\comfyui\custom_nodes\ComfyUI-MochiWrapper\mochi_preview\dit\joint_model\asymm_models_joint.py", line 703, in forward x, y_feat = block( File "E:\comfyui\python_embeded\lib\site-packages\torch\nn\modules\module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, kwargs) File "E:\comfyui\python_embeded\lib\site-packages\torch\nn\modules\module.py", line 1747, in _call_impl return forward_call(*args, *kwargs) File "E:\comfyui\comfyui\custom_nodes\ComfyUI-MochiWrapper\mochi_preview\dit\joint_model\asymm_models_joint.py", line 458, in forward x_attn, y_attn = self.attn( File "E:\comfyui\python_embeded\lib\site-packages\torch\nn\modules\module.py", line 1736, in _wrapped_call_impl return self._call_impl(args, kwargs) File "E:\comfyui\python_embeded\lib\site-packages\torch\nn\modules\module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "E:\comfyui\comfyui\custom_nodes\ComfyUI-MochiWrapper\mochi_preview\dit\joint_model\asymm_models_joint.py", line 354, in forward valid_token_indices=packed_indices["valid_token_indices_kv"], TypeError: 'NoneType' object is not subscriptable

[logtd]

If you update the ComfyUI-MochiWrapper repo and this repo that should fix the issue

[NeilWang079]

Thank you for your reply.