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

.

Browse Source
This commit is contained in:
pharmapsychotic
2023-09-09 20:29:39 -05:00
parent 2db78088a4
commit 661fbd95c4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/clip_interrogator_ext.py
View File

@@ -79,8 +79,8 @@ def unload():
print("Offloading CLIP Interrogator...")
ci.caption_model = ci.caption_model.to(devices.cpu)
ci.clip_model = ci.clip_model.to(devices.cpu)
ci.caption_offload = True
ci.clip_offload = True
ci.caption_offloaded = True
ci.clip_offloaded = True
devices.torch_gc()
def image_analysis(image, clip_model_name):