diff --git a/README.md b/README.md
index 17be96f..059f3d9 100644
--- a/README.md
+++ b/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.3
+pip install clip-interrogator==0.5.4
 ```
 
 You can then use it in your script
diff --git a/clip_interrogator/__init__.py b/clip_interrogator/__init__.py
index 27bb6f9..4317c31 100644
--- a/clip_interrogator/__init__.py
+++ b/clip_interrogator/__init__.py
@@ -1,4 +1,4 @@
 from .clip_interrogator import Interrogator, Config
 
-__version__ = '0.5.2'
+__version__ = '0.5.4'
 __author__ = 'pharmapsychotic'
\ No newline at end of file
diff --git a/setup.py b/setup.py
index b3297fc..81fcfef 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="clip-interrogator",
-    version="0.5.3",
+    version="0.5.4",
     license='MIT',
     author='pharmapsychotic',
     author_email='me@pharmapsychotic.com',