Comfy-Org / Comfy-Org/ComfyUI

WanVideoTinyVAELoader '解码器.7.conv.weight'

Open
#7,931 0 comments 0 reactions 0 assignees View on GitHub
Potential Bug
Dominant language
Python
Stars
133k
Forks
15.7k
Avg merge
1d 7h
Merged PRs (30d)
158

Description

### Expected Behavior

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 194, in _map_node_over_list
process_inputs(input_dict, i)
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 183, in process_inputs
results.append(getattr(obj, func)(**inputs))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\nodes.py", line 1002, in loadmodel
vae = TAEHV(vae_sd)
^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 177, in __init__
self.load_state_dict(self.patch_tgrow_layers(state_dict))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 190, in patch_tgrow_layers
if sd[key].shape[0] > new_sd[key].shape[0]:
~~^^^^^
KeyError: 'decoder.7.conv.weight'

### Actual Behavior

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 194, in _map_node_over_list
process_inputs(input_dict, i)
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 183, in process_inputs
results.append(getattr(obj, func)(**inputs))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\nodes.py", line 1002, in loadmodel
vae = TAEHV(vae_sd)
^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 177, in __init__
self.load_state_dict(self.patch_tgrow_layers(state_dict))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 190, in patch_tgrow_layers
if sd[key].shape[0] > new_sd[key].shape[0]:
~~^^^^^
KeyError: 'decoder.7.conv.weight'

### Steps to Reproduce

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 194, in _map_node_over_list
process_inputs(input_dict, i)
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 183, in process_inputs
results.append(getattr(obj, func)(**inputs))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\nodes.py", line 1002, in loadmodel
vae = TAEHV(vae_sd)
^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 177, in __init__
self.load_state_dict(self.patch_tgrow_layers(state_dict))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 190, in patch_tgrow_layers
if sd[key].shape[0] > new_sd[key].shape[0]:
~~^^^^^
KeyError: 'decoder.7.conv.weight'

### Debug Logs

```powershell
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 194, in _map_node_over_list
process_inputs(input_dict, i)
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 183, in process_inputs
results.append(getattr(obj, func)(**inputs))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\nodes.py", line 1002, in loadmodel
vae = TAEHV(vae_sd)
^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 177, in __init__
self.load_state_dict(self.patch_tgrow_layers(state_dict))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 190, in patch_tgrow_layers
if sd[key].shape[0] > new_sd[key].shape[0]:
~~^^^^^
KeyError: 'decoder.7.conv.weight'
```

### Other

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 194, in _map_node_over_list
process_inputs(input_dict, i)
File "D:\ComfyUI-aki-v1.6\ComfyUI\execution.py", line 183, in process_inputs
results.append(getattr(obj, func)(**inputs))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\nodes.py", line 1002, in loadmodel
vae = TAEHV(vae_sd)
^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 177, in __init__
self.load_state_dict(self.patch_tgrow_layers(state_dict))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v1.6\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper\taehv\taehv.py", line 190, in patch_tgrow_layers
if sd[key].shape[0] > new_sd[key].shape[0]:
~~^^^^^
KeyError: 'decoder.7.conv.weight'

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.