jhc13 / taggui

Tag manager and captioner for image datasets
GNU General Public License v3.0
727 stars 34 forks source link

Show space used by auto captioning models and allow to delete some of them #97

Open ChristianMayer opened 7 months ago

ChristianMayer commented 7 months ago

Trying an auto captioner downloads it automatically. This is a great feature!

But it comes with a few draw backs:

So I want to suggest a few enhancements:

madrooky commented 6 months ago

When you select a 7B or 13B model it is clear it is going to use some gigabytes. Not sure if its that easy to add the file size information.

I think it would be better having the list of models ready to use separated from the list of models that can be downloaded in the first place. Like an extra download tab. So annoying when accidentally choosing a model that is not on the SSD. Also the downloads should always go to the custom set path. I don't like cluttering an odd cache folder on my system drive with that kind of stuff.

yggdrasil75 commented 6 months ago

would also be nice to have a checkmark beside those that are already downloaded (maybe a doublecheck if currently loaded in memory)

Sana-A-E commented 4 months ago

The main issue I am experiencing is that I don't know which models I already have on disk and don't know where they are stored so I can't delete ones I don't use.

cwook commented 2 months ago

You can find the downloaded models on Windows at %userprofile%/cache/huggingface Personally I'd like to be able to move the location taggui downloads models to a different folder, since %userprofile% is locked to your windows install drive and I don't want such large models to be stored there.