Unable to unload DeepDanbooru model

toriato / stable-diffusion-webui-wd14-tagger

Labeling extension for Automatic1111's Web UI

1.34k stars 235 forks source link

Unable to unload DeepDanbooru model #33

Open toriato opened 1 year ago

toriato commented 1 year ago

Related issues:

https://github.com/keras-team/keras/issues/2102
https://github.com/toriato/stable-diffusion-webui-wd14-tagger/issues/31

AUTOMATIC1111 commented 1 year ago

Just in case you don't know, deepdanbooru is integrated into vanilla webui.

firefox_orvtuE68gt

According to the analysis of the person who initially implemented the model into webui, properly unloading tensorflow models is not possible; he had to start a new process and import tensorflow there. The implementation of deepdanbooru in webui uses torch.