From 5b6306667c5f32c97a62b922509a221381b22440 Mon Sep 17 00:00:00 2001
From: wnakano <nknvolpi@gmail.com>
Date: Sat, 4 Mar 2023 02:34:09 -0300
Subject: [PATCH] adding omegaconf and taming

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

diff --git a/setup.py b/setup.py
index e8c1b14..8e265b7 100644
--- a/setup.py
+++ b/setup.py
@@ -24,9 +24,11 @@
         "beartype",
         "einops>=0.6",
         "ema-pytorch",
+        "omegaconf>=2.3.0",
         "pillow",
         "sentencepiece",
         "torch>=1.6",
+        "traming>=0.0.1"
         "transformers",
         "torch>=1.6",
         "torchvision",