search

picobyte / stable-diffusion-webui-wd14-tagger

Labeling extension for Automatic1111's Web UI
539 stars 64 forks source link

Memory Leak #80

Closed lsycxyj closed 8 months ago

lsycxyj commented 8 months ago

Is there a way to cleanup the VRAM occupied by this extension? Or could you compute the tags in a subprocess and kill it to release the VRAM after the computation?

RoelKluin commented 8 months ago

It is not this extension, see #17

lsycxyj commented 8 months ago

@RoelKluin I know it's the problem related to tensorflow. However, is there a way that I can release the VRAM? I can't see the solution or workaround for this extension in #17.

RoelKluin commented 8 months ago

Not really. You can convert the model to onnx as WSH032 mentions there, otherwise restart the webui or avoid the tensorflow models.