Add omegaconf for tests (#3400)

Add omegaconfg

Author: patrickvonplaten

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

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

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