From 04bc1c0d77a9059a976f02bea9ca20871beab55d Mon Sep 17 00:00:00 2001
From: Korakoe <56580073+korakoe@users.noreply.github.com>
Date: Wed, 7 Jun 2023 13:57:31 +0800
Subject: [PATCH] Update setup.py

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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",