Open x-CK-x opened 10 months ago
the new models: https://huggingface.co/Thouph
I need to add to the configs, information pertaining to the file type:
https://huggingface.co/Thouph/eva02-clip-vit-large-7704/tree/main with:
https://huggingface.co/Thouph/experimental-eva02-vit-large-448-adventurous/tree/main with:
https://huggingface.co/Thouph/eva02-vit-large-448-8046/tree/main with:
https://huggingface.co/Thouph/experimental_efficientnetv2_m_8038/tree/main with:
https://huggingface.co/Thouph/experimental_efficientnetv2_m_8035/tree/main with:
(Additional ViT Model/s potentially available soon as well)
Working on a possible
way to automatically install CUDA in the same manner that it is done with auto1111.
The purpose of which would be to provide users the option of using their GPU.
balanced was chosen for now amongst all the effNet options: https://huggingface.co/Thouph/experimental_efficientnetv2_m_8035/blob/main/model_balanced.pth
and the following two additional models will be excluded for the time being: https://huggingface.co/Thouph/experimental_efficientnetv2_m_8038/tree/main https://huggingface.co/Thouph/experimental-eva02-vit-large-448-adventurous/tree/main
Current TODO
:
Create a generic handler for all the auto-tag/caption model/s using the configs just created.
GPU support: detect if GPU is available when user tries to enable GPU mode, if no gpu is present; then set mode back to cpu and display to user.
I need to make additional edits to the naming conventions of zack's model & tags file to reflect the consistencies of the new config file
code refactored to the following:
------- Model_Manager
/
/
/
GUI ------- Auto_Tag_Manager -------- Dataset_Manager
| \
| \
| \
| ------- Custom_File_Manager
|
|
|
-------- Meta_Data_Extractor
Several new model/s are available; The following must be done:
generic
handler for all the auto-tag/caption model/s