Lotus Depth workflow feeback

### Expected Behavior

Most image inputs can work properly, but some images will encounter errors.

### Actual Behavior


### Steps to Reproduce

### Input
[input.png.zip](https://github.com/user-attachments/files/20360236/input.png.zip)

### Workflow

[ComfyUI_00028_.json](https://github.com/user-attachments/files/20360240/ComfyUI_00028_.json)

### Debug Logs

```powershell
# ComfyUI Error Report
## Error Details
- **Node ID:** 8
- **Node Type:** VAEDecode
- **Exception Type:** RuntimeError
- **Exception Message:** Can't be indexed using 32-bit iterator
## Stack Trace

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/execution.py", line 349, 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 "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/execution.py", line 224, 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 "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/execution.py", line 196, in _map_node_over_list
process_inputs(input_dict, i)

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/execution.py", line 185, in process_inputs
results.append(getattr(obj, func)(**inputs))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/nodes.py", line 290, in decode
images = vae.decode(samples["samples"])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/comfy/sd.py", line 576, in decode
out = self.process_output(self.first_stage_model.decode(samples, **vae_options).to(self.output_device).float())
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/comfy/ldm/models/autoencoder.py", line 209, in decode
dec = self.decoder(dec, **decoder_kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/linmoumou/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1755, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/linmoumou/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1766, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/comfy/ldm/modules/diffusionmodules/model.py", line 719, in forward
h = self.up[i_level].block[i_block](h, temb, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/linmoumou/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1755, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/linmoumou/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1766, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/comfy/ldm/modules/diffusionmodules/model.py", line 188, in forward
h = self.norm1(h)
^^^^^^^^^^^^^

File "/Users/linmoumou/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1755, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/linmoumou/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1766, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/comfy/ops.py", line 142, in forward
return super().forward(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/linmoumou/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/modules/normalization.py", line 313, in forward
return F.group_norm(input, self.num_groups, self.weight, self.bias, self.eps)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/linmoumou/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/functional.py", line 2974, in group_norm
return torch.group_norm(
^^^^^^^^^^^^^^^^^

File "/Users/linmoumou/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/_refs/__init__.py", line 3234, in native_group_norm
out = (input_reshaped - mean) * rstd
~~~~~~~~~~~~~~~^~~~~~

## System Information
- **ComfyUI Version:** 0.3.34
- **Arguments:** /Applications/ComfyUI.app/Contents/Resources/ComfyUI/main.py --user-directory /Users/linmoumou/Documents/ComfyUI/user --input-directory /Users/linmoumou/Documents/ComfyUI/input --output-directory /Users/linmoumou/Documents/ComfyUI/output --front-end-root /Applications/ComfyUI.app/Contents/Resources/ComfyUI/web_custom_versions/desktop_app --base-directory /Users/linmoumou/Documents/ComfyUI --extra-model-paths-config /Users/linmoumou/Library/Application Support/ComfyUI/extra_models_config.yaml --log-stdout --listen 0.0.0.0 --port 8000
- **OS:** posix
- **Python Version:** 3.12.9 (main, Feb 12 2025, 15:09:19) [Clang 19.1.6 ]
- **Embedded Python:** false
- **PyTorch Version:** 2.8.0.dev20250514
## Devices

- **Name:** mps
- **Type:** mps
- **VRAM Total:** 38654705664
- **VRAM Free:** 11554717696
- **Torch VRAM Total:** 38654705664
- **Torch VRAM Free:** 11554717696

## Logs

2025-05-21T12:43:35.845144 - Adding extra search path custom_nodes /Users/linmoumou/Documents/ComfyUI/custom_nodes
2025-05-21T12:43:35.845189 - Adding extra search path download_model_base /Users/linmoumou/Documents/ComfyUI/models
2025-05-21T12:43:35.845209 - Adding extra search path custom_nodes /Applications/ComfyUI.app/Contents/Resources/ComfyUI/custom_nodes
2025-05-21T12:43:35.845222 - Setting output directory to: /Users/linmoumou/Documents/ComfyUI/output
2025-05-21T12:43:35.845236 - Setting input directory to: /Users/linmoumou/Documents/ComfyUI/input
2025-05-21T12:43:35.845244 - Setting user directory to: /Users/linmoumou/Documents/ComfyUI/user
2025-05-21T12:43:36.032335 - [START] Security scan2025-05-21T12:43:36.032345 -
2025-05-21T12:43:36.288804 - [DONE] Security scan2025-05-21T12:43:36.288839 -
2025-05-21T12:43:36.350208 - ## ComfyUI-Manager: installing dependencies done.2025-05-21T12:43:36.350251 -
2025-05-21T12:43:36.350274 - ** ComfyUI startup time:2025-05-21T12:43:36.350284 - 2025-05-21T12:43:36.350293 - 2025-05-21 12:43:36.3502025-05-21T12:43:36.350301 -
2025-05-21T12:43:36.350313 - ** Platform:2025-05-21T12:43:36.350320 - 2025-05-21T12:43:36.350327 - Darwin2025-05-21T12:43:36.350334 -
2025-05-21T12:43:36.350342 - ** Python version:2025-05-21T12:43:36.350349 - 2025-05-21T12:43:36.350356 - 3.12.9 (main, Feb 12 2025, 15:09:19) [Clang 19.1.6 ]2025-05-21T12:43:36.350363 -
2025-05-21T12:43:36.350369 - ** Python executable:2025-05-21T12:43:36.350376 - 2025-05-21T12:43:36.350383 - /Users/linmoumou/Documents/ComfyUI/.venv/bin/python2025-05-21T12:43:36.350389 -
2025-05-21T12:43:36.350396 - ** ComfyUI Path:2025-05-21T12:43:36.350402 - 2025-05-21T12:43:36.350408 - /Applications/ComfyUI.app/Contents/Resources/ComfyUI2025-05-21T12:43:36.350415 -
2025-05-21T12:43:36.350421 - ** ComfyUI Base Folder Path:2025-05-21T12:43:36.350428 - 2025-05-21T12:43:36.350434 - /Applications/ComfyUI.app/Contents/Resources/ComfyUI2025-05-21T12:43:36.350440 -
2025-05-21T12:43:36.350447 - ** User directory:2025-05-21T12:43:36.350453 - 2025-05-21T12:43:36.350459 - /Users/linmoumou/Documents/ComfyUI/user2025-05-21T12:43:36.350465 -
2025-05-21T12:43:36.350472 - ** ComfyUI-Manager config path:2025-05-21T12:43:36.350478 - 2025-05-21T12:43:36.350484 - /Users/linmoumou/Documents/ComfyUI/user/default/ComfyUI-Manager/config.ini2025-05-21T12:43:36.350490 -
2025-05-21T12:43:36.350499 - ** Log path:2025-05-21T12:43:36.350505 - 2025-05-21T12:43:36.350511 - /Users/linmoumou/Documents/ComfyUI/user/comfyui.log2025-05-21T12:43:36.350517 -
2025-05-21T12:43:36.621093 - [ComfyUI-Manager] Failed to restore comfyui-frontend-package
2025-05-21T12:43:36.621217 - expected str, bytes or os.PathLike object, not NoneType
2025-05-21T12:43:36.622662 -
Prestartup times for custom nodes:
2025-05-21T12:43:36.622712 - 0.8 seconds: /Applications/ComfyUI.app/Contents/Resources/ComfyUI/custom_nodes/ComfyUI-Manager
2025-05-21T12:43:36.622734 -
2025-05-21T12:43:37.328476 - Checkpoint files will always be loaded safely.
2025-05-21T12:43:37.349945 - Total VRAM 36864 MB, total RAM 36864 MB
2025-05-21T12:43:37.350013 - pytorch version: 2.8.0.dev20250514
2025-05-21T12:43:37.351039 - Mac Version (15, 1)
2025-05-21T12:43:37.351132 - Set vram state to: SHARED
2025-05-21T12:43:37.351160 - Device: mps
2025-05-21T12:43:37.926731 - Using sub quadratic optimization for attention, if you have memory or speed issues try using: --use-split-cross-attention
2025-05-21T12:43:38.822104 - Python version: 3.12.9 (main, Feb 12 2025, 15:09:19) [Clang 19.1.6 ]
2025-05-21T12:43:38.822180 - ComfyUI version: 0.3.34
2025-05-21T12:43:38.844283 - [Prompt Server] web root: /Applications/ComfyUI.app/Contents/Resources/ComfyUI/web_custom_versions/desktop_app
2025-05-21T12:43:39.876775 -
2025-05-21T12:43:39.876805 - [rgthree-comfy] Loaded 46 magnificent nodes. 🎉2025-05-21T12:43:39.876817 -
2025-05-21T12:43:39.876827 -
2025-05-21T12:43:39.878763 - Total VRAM 36864 MB, total RAM 36864 MB
2025-05-21T12:43:39.878815 - pytorch version: 2.8.0.dev20250514
2025-05-21T12:43:39.878931 - Mac Version (15, 1)
2025-05-21T12:43:39.879007 - Set vram state to: SHARED
2025-05-21T12:43:39.879031 - Device: mps
2025-05-21T12:43:39.881734 - OpenCV not installed2025-05-21T12:43:39.881756 -
2025-05-21T12:43:39.885297 - Traceback (most recent call last):
File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/nodes.py", line 2131, in load_custom_node
module_spec.loader.exec_module(module)
File "", line 999, in exec_module
File "", line 488, in _call_with_frames_removed
File "/Users/linmoumou/Documents/ComfyUI/custom_nodes/comfyui-kjnodes/__init__.py", line 7, in
from .nodes.mask_nodes import *
File "/Users/linmoumou/Documents/ComfyUI/custom_nodes/comfyui-kjnodes/nodes/mask_nodes.py", line 1274, in
import cv2
ModuleNotFoundError: No module named 'cv2'

2025-05-21T12:43:39.885420 - Cannot import /Users/linmoumou/Documents/ComfyUI/custom_nodes/comfyui-kjnodes module for custom nodes: No module named 'cv2'
2025-05-21T12:43:39.889129 - ### Loading: ComfyUI-Manager (V3.30.4)
2025-05-21T12:43:39.889325 - [ComfyUI-Manager] network_mode: public
2025-05-21T12:43:39.905802 - ### ComfyUI Revision: UNKNOWN (The currently installed ComfyUI is not a Git repository)
2025-05-21T12:43:39.911738 -
Import times for custom nodes:
2025-05-21T12:43:39.911796 - 0.0 seconds: /Applications/ComfyUI.app/Contents/Resources/ComfyUI/custom_nodes/websocket_image_save.py
2025-05-21T12:43:39.911819 - 0.0 seconds: /Users/linmoumou/Documents/ComfyUI/custom_nodes/ComfyUI-Align
2025-05-21T12:43:39.911839 - 0.0 seconds: /Users/linmoumou/Documents/ComfyUI/custom_nodes/rgthree-comfy
2025-05-21T12:43:39.912367 - 0.0 seconds (IMPORT FAILED): /Users/linmoumou/Documents/ComfyUI/custom_nodes/comfyui-kjnodes
2025-05-21T12:43:39.912389 - 0.0 seconds: /Applications/ComfyUI.app/Contents/Resources/ComfyUI/custom_nodes/ComfyUI-Manager
2025-05-21T12:43:39.912484 - 0.7 seconds: /Users/linmoumou/Documents/ComfyUI/custom_nodes/ace-step
2025-05-21T12:43:39.912874 -
2025-05-21T12:43:39.919657 - Starting server

2025-05-21T12:43:39.919953 - To see the GUI go to: http://0.0.0.0:8000
2025-05-21T12:43:41.349510 - [ERROR] An error occurred while retrieving information for the 'ACELoRALoader' node.
2025-05-21T12:43:41.350875 - Traceback (most recent call last):
File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/server.py", line 595, in get_object_info
out[x] = node_info(x)
^^^^^^^^^^^^
File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/server.py", line 559, in node_info
info['input'] = obj_class.INPUT_TYPES()
^^^^^^^^^^^^^^^^^^^^^^^
File "/Users/linmoumou/Documents/ComfyUI/custom_nodes/ace-step/ace_step_nodes.py", line 322, in INPUT_TYPES
models = [name for name in os.listdir(loras_path) if not name.startswith(".")]
^^^^^^^^^^^^^^^^^^^^^^
FileNotFoundError: [Errno 2] No such file or directory: '/Users/linmoumou/Documents/ComfyUI/models/TTS/ACE-Step-v1-3.5B/loras'

2025-05-21T12:43:41.350928 - [ERROR] An error occurred while retrieving information for the 'ACEModelLoader' node.
2025-05-21T12:43:41.351012 - Traceback (most recent call last):
File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/server.py", line 595, in get_object_info
out[x] = node_info(x)
^^^^^^^^^^^^
File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/server.py", line 559, in node_info
info['input'] = obj_class.INPUT_TYPES()
^^^^^^^^^^^^^^^^^^^^^^^
File "/Users/linmoumou/Documents/ComfyUI/custom_nodes/ace-step/ace_step_nodes.py", line 189, in INPUT_TYPES
models = [name for name in os.listdir(model_path) if os.path.isdir(os.path.join(model_path, name)) and not name.startswith(".")]
^^^^^^^^^^^^^^^^^^^^^^
FileNotFoundError: [Errno 2] No such file or directory: '/Users/linmoumou/Documents/ComfyUI/models/TTS/ACE-Step-v1-3.5B'

2025-05-21T12:43:41.809725 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/model-list.json
2025-05-21T12:43:41.997870 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/alter-list.json
2025-05-21T12:43:42.349051 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/github-stats.json
2025-05-21T12:43:43.600078 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/custom-node-list.json
2025-05-21T12:43:43.937002 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/extension-node-map.json
2025-05-21T12:43:57.374140 - FETCH ComfyRegistry Data: 5/862025-05-21T12:43:57.374324 -
2025-05-21T12:44:10.825738 - FETCH ComfyRegistry Data: 10/862025-05-21T12:44:10.825922 -
2025-05-21T12:44:24.710356 - FETCH ComfyRegistry Data: 15/862025-05-21T12:44:24.710550 -
2025-05-21T12:44:34.148529 - FETCH ComfyRegistry Data: 20/862025-05-21T12:44:34.148567 -
2025-05-21T12:44:35.382902 - got prompt
2025-05-21T12:44:35.411623 - Using split attention in VAE
2025-05-21T12:44:35.412502 - Using split attention in VAE
2025-05-21T12:44:35.578932 - VAE load device: mps, offload device: cpu, dtype: torch.bfloat16
2025-05-21T12:44:35.617801 - Requested to load AutoencoderKL
2025-05-21T12:44:35.690047 - loaded completely 9.5367431640625e+25 159.55708122253418 True
2025-05-21T12:44:36.799576 - input_blocks.1.0.skip_connection.weight down_blocks.0.resnets.0.conv_shortcut.weight
2025-05-21T12:44:36.799661 - input_blocks.1.0.skip_connection.bias down_blocks.0.resnets.0.conv_shortcut.bias
2025-05-21T12:44:36.799693 - input_blocks.2.0.skip_connection.weight down_blocks.0.resnets.1.conv_shortcut.weight
2025-05-21T12:44:36.799714 - input_blocks.2.0.skip_connection.bias down_blocks.0.resnets.1.conv_shortcut.bias
2025-05-21T12:44:36.799741 - input_blocks.5.0.skip_connection.weight down_blocks.1.resnets.1.conv_shortcut.weight
2025-05-21T12:44:36.799758 - input_blocks.5.0.skip_connection.bias down_blocks.1.resnets.1.conv_shortcut.bias
2025-05-21T12:44:36.799792 - input_blocks.8.0.skip_connection.weight down_blocks.2.resnets.1.conv_shortcut.weight
2025-05-21T12:44:36.799807 - input_blocks.8.0.skip_connection.bias down_blocks.2.resnets.1.conv_shortcut.bias
2025-05-21T12:44:36.799825 - input_blocks.10.0.skip_connection.weight down_blocks.3.resnets.0.conv_shortcut.weight
2025-05-21T12:44:36.799838 - input_blocks.10.0.skip_connection.bias down_blocks.3.resnets.0.conv_shortcut.bias
2025-05-21T12:44:36.799852 - input_blocks.11.0.skip_connection.weight down_blocks.3.resnets.1.conv_shortcut.weight
2025-05-21T12:44:36.799868 - input_blocks.11.0.skip_connection.bias down_blocks.3.resnets.1.conv_shortcut.bias
2025-05-21T12:44:36.799881 - input_blocks.12.0.op.weight down_blocks.3.downsamplers.0.conv.weight
2025-05-21T12:44:36.799894 - input_blocks.12.0.op.bias down_blocks.3.downsamplers.0.conv.bias
2025-05-21T12:44:36.799912 - middle_block.0.skip_connection.weight mid_block.resnets.0.conv_shortcut.weight
2025-05-21T12:44:36.799926 - middle_block.0.skip_connection.bias mid_block.resnets.0.conv_shortcut.bias
2025-05-21T12:44:36.799940 - middle_block.2.skip_connection.weight mid_block.resnets.1.conv_shortcut.weight
2025-05-21T12:44:36.799953 - middle_block.2.skip_connection.bias mid_block.resnets.1.conv_shortcut.bias
2025-05-21T12:44:36.800004 - output_blocks.11.2.conv.weight up_blocks.3.upsamplers.0.conv.weight
2025-05-21T12:44:36.800017 - output_blocks.11.2.conv.bias up_blocks.3.upsamplers.0.conv.bias
2025-05-21T12:44:36.800030 - label_emb.0.2.bias add_embedding.linear_2.bias
2025-05-21T12:44:36.800042 - label_emb.0.0.bias add_embedding.linear_1.bias
2025-05-21T12:44:36.800063 - label_emb.0.0.weight add_embedding.linear_1.weight
2025-05-21T12:44:36.800082 - label_emb.0.2.weight add_embedding.linear_2.weight
2025-05-21T12:44:36.807478 - model weight dtype torch.float16, manual cast: None
2025-05-21T12:44:36.807758 - model_type IMG_TO_IMG
2025-05-21T12:44:38.322334 - Warning: TAESD previews enabled, but could not find models/vae_approx/taesd_decoder
2025-05-21T12:44:38.322805 - Requested to load Lotus
2025-05-21T12:44:38.607640 - loaded completely 9.5367431640625e+25 1654.733283996582 True
2025-05-21T12:44:41.160191 -
100%|██████████| 1/1 [00:02<00:00, 2.50s/it]2025-05-21T12:44:41.166977 -
100%|██████████| 1/1 [00:02<00:00, 2.51s/it]2025-05-21T12:44:41.166999 -
2025-05-21T12:44:42.736124 - Prompt executed in 7.35 seconds
2025-05-21T12:44:43.701286 - FETCH ComfyRegistry Data: 25/862025-05-21T12:44:43.701496 -
2025-05-21T12:44:49.836055 - got prompt
2025-05-21T12:44:51.205528 - Warning: TAESD previews enabled, but could not find models/vae_approx/taesd_decoder
2025-05-21T12:44:51.378212 -
0%| | 0/1 [00:00