Fix PuLID VRAM leak - Githubissues

Mikubill / sd-webui-controlnet

WebUI extension for ControlNet

GNU General Public License v3.0

16.42k stars 1.91k forks source link

Fix PuLID VRAM leak #2902

Closed huchenlei closed 1 month ago

huchenlei commented 1 month ago

Closes https://github.com/Mikubill/sd-webui-controlnet/issues/2862.

Run torch.cuda.empty_cache() to clear gpu cache.

Refrence in original PuLID repo: https://github.com/ToTheBeginning/PuLID/blob/7572f73de403187699075504e8415f9b00d98858/pulid/pipeline.py#L95-L96