Support for New Auto-Tagging/Captioning Model/s Available

[x-CK-x]

Several new model/s are available; The following must be done:

• add them to the download tab
• create a config for each model
• create a generic handler for all the auto-tag/caption model/s
• GPU support (optional)
• protobuffer (.pb) to .onnx conversion to use the (onnx-runtime library)

[x-CK-x]

• CogVLM might need a "newer" interface to work successfully

[x-CK-x]

the new models: https://huggingface.co/Thouph

[x-CK-x]

I need to add to the configs, information pertaining to the file type:

[x-CK-x]

https://huggingface.co/Thouph/eva02-clip-vit-large-7704/tree/main with:

• https://huggingface.co/Thouph/eva02-clip-vit-large-7704/blob/main/model.onnx
• https://huggingface.co/Thouph/eva02-clip-vit-large-7704/blob/main/model.pth
• https://huggingface.co/Thouph/eva02-clip-vit-large-7704/blob/main/tags.json

https://huggingface.co/Thouph/experimental-eva02-vit-large-448-adventurous/tree/main with:

• https://huggingface.co/Thouph/experimental-eva02-vit-large-448-adventurous/blob/main/model.pth

https://huggingface.co/Thouph/eva02-vit-large-448-8046/tree/main with:

• https://huggingface.co/Thouph/eva02-vit-large-448-8046/blob/main/model.pth
• https://huggingface.co/Thouph/eva02-vit-large-448-8046/blob/main/tags_8041.json

https://huggingface.co/Thouph/experimental_efficientnetv2_m_8038/tree/main with:

• https://huggingface.co/Thouph/experimental_efficientnetv2_m_8038/blob/main/model.pth

https://huggingface.co/Thouph/experimental_efficientnetv2_m_8035/tree/main with:

• https://huggingface.co/Thouph/experimental_efficientnetv2_m_8035/blob/main/model_adventurous.pth
• https://huggingface.co/Thouph/experimental_efficientnetv2_m_8035/blob/main/model_balanced.pth
• https://huggingface.co/Thouph/experimental_efficientnetv2_m_8035/blob/main/model_conservative.pth
• https://huggingface.co/Thouph/experimental_efficientnetv2_m_8035/blob/main/model_mix_step42000.pth
• https://huggingface.co/Thouph/experimental_efficientnetv2_m_8035/blob/main/tags_8034.json

(Additional ViT Model/s potentially available soon as well)

[x-CK-x]

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.

[x-CK-x]

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

[x-CK-x]

6306c1b

[x-CK-x]

5adc7aa

[x-CK-x]

8cb8d12

[x-CK-x]

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.

[x-CK-x]

#32

[x-CK-x]

33ab23a

[x-CK-x]

50255a6

[x-CK-x]

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

[x-CK-x]

code refactored to the following:

                                   ------- Model_Manager
                                 /
                               /
                             /
GUI ------- Auto_Tag_Manager -------- Dataset_Manager
                    |        \
                    |          \
                    |            \
                    |              ------- Custom_File_Manager
                    |              
                    |              
                    |              
                     -------- Meta_Data_Extractor