Add omegaconf for tests (huggingface#3400)

Add omegaconfg
hari10599 · May 20, 2023 · 8b66534 · 8b66534
1 parent 3bab713
commit 8b66534
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile
@@ -37,6 +37,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \
         numpy \
         scipy \
         tensorboard \
-        transformers
+        transformers \
+        omegaconf
 
 CMD ["/bin/bash"]
diff --git a/setup.py b/setup.py
@@ -96,6 +96,7 @@
     "k-diffusion>=0.0.12",
     "librosa",
     "numpy",
+    "omegaconf",
     "parameterized",
     "protobuf>=3.20.3,<4",
     "pytest",
@@ -191,6 +192,7 @@ def run(self):
     "Jinja2",
     "k-diffusion",
     "librosa",
+    "omegaconf",
     "parameterized",
     "pytest",
     "pytest-timeout",

diff --git a/src/diffusers/dependency_versions_table.py b/src/diffusers/dependency_versions_table.py
@@ -20,6 +20,7 @@
     "k-diffusion": "k-diffusion>=0.0.12",
     "librosa": "librosa",
     "numpy": "numpy",
+    "omegaconf": "omegaconf",
     "parameterized": "parameterized",
     "protobuf": "protobuf>=3.20.3,<4",
     "pytest": "pytest",