Skip to content

[Feature Request]: Unload ReActor Models and VRAM Debug #8

@Taruvi

Description

@Taruvi

I've had a problem that's bugged me for a long, long time to do with VRAM being clogged by something. I've tracked it down to ReActor. Whenever I've used it, it would keep about 2.5GB VRAM clogged up until I completely restarted Swarm/Comfy.

It got to the point where I've only been using ReActor sparingly, to the point of hardly ever really using it anymore.

In fact, I reported it in the Swarm repo way back. I think around that time I was doing lots of generating images, then doing I2V in WAN, and that's why ReActor was always taking up that annoying little bit of VRAM all the time that would give me constant OOMs until I fully restarted everything.

However...

I did a little bit of digging around and I see there's been a node there this whole time that's perfect for fixing this problem:

Image

I've been testing this in a Comfy Workflow and it does the job exactly as the name implies.

Further to this, I have it hooked up to KJ's VRAM Debug node:

Image

With both combined, my VRAM usage drops to next to nothing after every generation.

VRAM Debug works very well in an extension I currently use a lot: SwarmUI SeedVR2 Upscaler Extension.

The readme for that says:

VRAM Cleanup: Automatically unloads the main generation model before upscaling (requires KJNodes)

It works great.

Considering Unload ReActor Models is already included in ReActor anyway, and VRAM Debug is part of KJ's Nodes which I suppose a lot of people have installed already, my feature request is:

  1. Add a toggle in the extension for the Unload ReActor Models node to run it after every generation
  2. Add a toggle in the extension for the VRAM Debug node to run after every generation

Every little bit helps for those of us stuck with what we have due to stupid GPU prices.

Thanks.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions