Add omegaconf for tests (#3400)

Add omegaconfg

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 @@ _deps = [
     "k-diffusion>=0.0.12",
     "librosa",
     "numpy",
+    "omegaconf",
     "parameterized",
     "protobuf>=3.20.3,<4",
     "pytest",
@@ -191,6 +192,7 @@ extras["test"] = deps_list(
     "Jinja2",
     "k-diffusion",
     "librosa",
+    "omegaconf",
     "parameterized",
     "pytest",
     "pytest-timeout",

src/diffusers/dependency_versions_table.py

@@ -20,6 +20,7 @@ deps = {
     "k-diffusion": "k-diffusion>=0.0.12",
     "librosa": "librosa",
     "numpy": "numpy",
+    "omegaconf": "omegaconf",
     "parameterized": "parameterized",
     "protobuf": "protobuf>=3.20.3,<4",
     "pytest": "pytest",