gitmylo / audio-webui

A webui for different audio related Neural Networks
MIT License
1.01k stars 94 forks source link

[BUG REPORT] Memory leak / out of memory error #74

Closed jh-792 closed 1 year ago

jh-792 commented 1 year ago

When generating text to speech using Bark, memory never seems to be freed up.

After 20-30 generations (whether in a single long chunk of text that gets split up, or as many individual generations) my 32GB of memory has been gobbled up causing the system to lock up.

Unloading Bark and reloading the model doesn't free up the memory, the only way to free up the memory seems to be to close and reload audio-webui.

I'm running Ubuntu 22.04.

gitmylo commented 1 year ago

That's odd, i've been able to perform at least 50 generations on bark, i have 48gb ram, and am running Kubuntu 23.04. I'll see if i have missed this maybe.

jh-792 commented 1 year ago

Thanks, I've actually just run into the same issue with another implementation of Bark, so I suspect it's an issue with Bark itself rather than this project... Happy to run any tests that might help narrow it down though.

gitmylo commented 1 year ago

image

Yeah there appears to be a memory leak, i'll tell the bark devs about this. (Closing this issue as it appears related to bark)

gitmylo commented 1 year ago

It could also be an issue with gradio, maybe