From 661fbd95c4a4483664cd29ab4e51505f8e3035d1 Mon Sep 17 00:00:00 2001 From: pharmapsychotic Date: Sat, 9 Sep 2023 20:29:39 -0500 Subject: [PATCH] . --- scripts/clip_interrogator_ext.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/clip_interrogator_ext.py b/scripts/clip_interrogator_ext.py index 4f7ded1..b1763ec 100644 --- a/scripts/clip_interrogator_ext.py +++ b/scripts/clip_interrogator_ext.py @@ -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):