diff --git a/README.md b/README.md
index 28d5c2c..7dcefc8 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.0
+pip install clip-interrogator==0.5.1
 ```
 
 You can then use it in your script
diff --git a/clip_interrogator/__init__.py b/clip_interrogator/__init__.py
index 0992c67..28fc2c6 100644
--- a/clip_interrogator/__init__.py
+++ b/clip_interrogator/__init__.py
@@ -1,4 +1,4 @@
 from .clip_interrogator import Interrogator, Config
 
-__version__ = '0.5.0'
+__version__ = '0.5.1'
 __author__ = 'pharmapsychotic'
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 918cef9..4b676f7 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="clip-interrogator",
-    version="0.5.0",
+    version="0.5.1",
     license='MIT',
     author='pharmapsychotic',
     author_email='me@pharmapsychotic.com',