shiimizu / ComfyUI_smZNodes

Custom nodes for ComfyUI such as CLIP Text Encode++
GNU General Public License v3.0
190 stars 14 forks source link

gpu-only fix #2

Closed pamparamm closed 1 year ago

pamparamm commented 1 year ago

CLIP Text Encode++ works fine when using ComfyUI with --normalvram flag, but fails with --gpu-only due to mismatch between devices. This PR fixes that issue.

shiimizu commented 1 year ago

Thanks for the PR. While it does solve the issue, the underlying problem was that self.device was always on cpu.

pamparamm commented 1 year ago

Yep, everything works with all vram arguments after 15626f2.