huchenlei / ComfyUI-layerdiffuse

Layer Diffuse custom nodes
Apache License 2.0
1.44k stars 142 forks source link

Fix decode output device/dtype #29

Closed huchenlei closed 6 months ago

huchenlei commented 6 months ago

Closes #28.

This PR makes sure that after the decode, the output tensor's dtype/device matches the input pixel tensor's dtype/device.