Memory Leak

Question

Memory Leak

lsycxyj opened this issue a year ago · 3 comments

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?

Answer 1 · 2023-10-19T11:36:28.000Z

It is not this extension, see #17

Answer 2 · 2023-10-27T07:21:57.000Z

@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.

Answer 3 · 2023-10-27T08:54:18.000Z

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