diff --git a/setup.py b/setup.py
index dbaf0fe..f55dfee 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="muse-maskgit-pytorch",
     packages=find_packages(exclude=[]),
-    version="0.1.0",
+    version="0.1.6",
     license="MIT",
     description="MUSE - Text-to-Image Generation via Masked Generative Transformers, in Pytorch",
     author="Phil Wang",
@@ -39,6 +39,7 @@
         "tqdm-loggable",
         "vector-quantize-pytorch>=0.10.14",
         "lion-pytorch",
+        "omegaconf"
     ],
     classifiers=[
         "Development Status :: 4 - Beta",