Minor fix to BLIP offloading

README.md

@@ -36,7 +36,7 @@ Install with PIP
 pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117
 
 # install clip-interrogator
-pip install clip-interrogator==0.5.2
+pip install clip-interrogator==0.5.3
 ```
 
 You can then use it in your script

clip_interrogator/clip_interrogator.py

@@ -300,7 +300,7 @@ class Interrogator():
             self.blip_offloaded = False
 
     def _prepare_clip(self):
-        if self.config.blip_offload:
+        if self.config.blip_offload and not self.blip_offloaded:
             self.blip_model = self.blip_model.to('cpu')
             self.blip_offloaded = True
         if self.clip_offloaded:

setup.py

@@ -5,7 +5,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="clip-interrogator",
-    version="0.5.2",
+    version="0.5.3",
     license='MIT',
     author='pharmapsychotic',
     author_email='me@pharmapsychotic.com',