adding omegaconf and taming

ZeroCool940711 · Mar 4, 2023 · 5b63066 · 5b63066
1 parent ea0e421
commit 5b63066
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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",