ComfyUI nodes released

[set-soft]

Hi @qianyu-dlut !

My nodes, which support DiffDIS, are now available to download from the ComfyUI Manager. The name is: "Remove Background (SET)". So people can now install it from inside ComfyUI.

Of course the nodes can also be manually installed from its repo: ComfyUI-RemoveBackground_SET

Models are automatically downloaded to the ComfyUI/models/rembg folder.

For DiffDIS you can choose to download:

• https://huggingface.co/set-soft/RemBG/resolve/main/DiffDIS/DiffDIS_F16.safetensors
• https://huggingface.co/set-soft/RemBG/resolve/main/DiffDIS/DiffDIS_F32.safetensors

I repacked them to include the "empty text conditional", provide a 16 bits version and include some metadata. These are based on the file on Google Drive. You can see the code I used here: diffdis_repack.py.

Do you know about other pretrained checkpoints?

If you have some time please take a look at the project README and tell me if I committed some error about DiffDIS. As DiffDIS needs the SD Turbo VAE and also generates edges I added extra examples to show how to use the model.

Also: if you have the images used in the paper I suggest uploading them to your repo, so I can add a workflow example that downloads the images from your repo and uses them to show DiffDIS working.

Note that my nodes can help you to compare your work with others, you can generate things like this:

[qianyu-dlut]

Hi @set-soft , really appreciate the ComfyUI Manager support and the detailed DiffDIS notes. I’ll review when I can and reach out if anything comes up. Many thanks!