Update setup.py

ZeroCool940711 · Jun 7, 2023 · 04bc1c0 · 04bc1c0
1 parent 729cbad
commit 04bc1c0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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",