hiddenswitch / ComfyUI

A powerful and modular stable diffusion GUI with a graph/nodes interface.
GNU General Public License v3.0
30 stars 10 forks source link

Loading a TAESD as your VAE throws an error #5

Closed dyvoid closed 4 months ago

dyvoid commented 4 months ago

I'm running this on Windows through WSL (Ubuntu) and I seem to have a problem with the Tiny AutoEncoders. These are the high performance VAEs that are installed in the models/vae_approx directory (taesd and taesdxl). Here is the error I'm getting:

Error occurred when executing VAELoader:

'dict' object has no attribute 'VAE'

File "/mnt/g/Projects/comfy-wsl-test/.venv/lib/python3.10/site-packages/comfy/cmd/execution.py", line 183, in recursive_execute
output_data, output_ui = get_output_data(obj, input_data_all)
File "/mnt/g/Projects/comfy-wsl-test/.venv/lib/python3.10/site-packages/comfy/cmd/execution.py", line 102, in get_output_data
return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True)
File "/mnt/g/Projects/comfy-wsl-test/.venv/lib/python3.10/site-packages/comfy/cmd/execution.py", line 95, in map_node_over_list
results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
File "/mnt/g/Projects/comfy-wsl-test/.venv/lib/python3.10/site-packages/comfy/nodes/base_nodes.py", line 704, in load_vae
vae = sd.VAE(sd=sd)
doctorpangloss commented 4 months ago

Thanks for the report, your issue should be resolved in master now.

dyvoid commented 4 months ago

I tested it and can confirm it works. Many thanks for the quick turnaround, I appreciate it!

doctorpangloss commented 4 months ago

previews are now enabled over distributed workers too