From 68460a19ebfece6ff9476e42201f94f0498cb9c7 Mon Sep 17 00:00:00 2001 From: bolshoytoster Date: Sun, 12 Feb 2023 17:03:57 +0000 Subject: [PATCH] Minor bug fix --- clip_interrogator/clip_interrogator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/clip_interrogator/clip_interrogator.py b/clip_interrogator/clip_interrogator.py index 1d8899e..30545a8 100644 --- a/clip_interrogator/clip_interrogator.py +++ b/clip_interrogator/clip_interrogator.py @@ -440,6 +440,7 @@ class Interrogator: if self.blip_loaded: # Move CLIP to intended device self.blip_model.to("cpu") self.cli_model.to(self.device) + self.blip_loaded = False image_features = self.image_to_features(image) flaves = self.flavors.rank(