pharmapsychotic/clip-interrogator-ext
1
0
Fork 0
mirror of https://github.com/pharmapsychotic/clip-interrogator-ext.git synced 2026-04-26 17:29:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix analyze tab

Browse Source
This commit is contained in:
pharmapsychotic
2023-03-19 23:54:28 -05:00
parent 6b950289b8
commit f1aa3d6d8a
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/clip_interrogator_ext.py
View File

@@ -11,7 +11,7 @@ from clip_interrogator import Config, Interrogator, list_caption_models, list_cl
from modules import devices, lowvram, script_callbacks, shared
__version__ = '0.1.0'
__version__ = '0.1.1'
ci = None
low_vram = False
@@ -56,14 +56,15 @@ def load(clip_model_name, caption_model_name):
print(f"Loading CLIP Interrogator {clip_interrogator.__version__}...")
config = Config(
device=devices.get_optimal_device(),
clip_model_name=clip_model_name,
caption_model_name=caption_model_name
clip_model_name=clip_model_name
)
if caption_model_name:
config.caption_model_name = caption_model_name
config.cache_path = 'models/clip-interrogator'
if low_vram:
config.apply_low_vram_defaults()
ci = Interrogator(config)
if caption_model_name != ci.config.caption_model_name:
if caption_model_name and caption_model_name != ci.config.caption_model_name:
ci.config.caption_model_name = caption_model_name
ci.load_caption_model()
if clip_model_name != ci.config.clip_model_name:
@@ -81,7 +82,7 @@ def unload():
devices.torch_gc()
def image_analysis(image, clip_model_name):
load(clip_model_name)
load(clip_model_name, None)
image = image.convert('RGB')
image_features = ci.image_to_features(image)