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(