Skip to content

GitLab

Unverified Commit 58c6f9cb authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Add omegaconf for tests (#3400) 

Add omegaconfg
parent af2a2376
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
docker/diffusers-pytorch-cuda/Dockerfile
View file @ 58c6f9cb
...@@ -37,6 +37,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \ ...@@ -37,6 +37,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \
numpy \ numpy \
scipy \ scipy \
tensorboard \ tensorboard \
transformers transformers \
omegaconf
CMD ["/bin/bash"] CMD ["/bin/bash"]
setup.py
View file @ 58c6f9cb
...@@ -96,6 +96,7 @@ _deps = [ ...@@ -96,6 +96,7 @@ _deps = [
"k-diffusion>=0.0.12", "k-diffusion>=0.0.12",
"librosa", "librosa",
"numpy", "numpy",
"omegaconf",
"parameterized", "parameterized",
"protobuf>=3.20.3,<4", "protobuf>=3.20.3,<4",
"pytest", "pytest",
...@@ -191,6 +192,7 @@ extras["test"] = deps_list( ...@@ -191,6 +192,7 @@ extras["test"] = deps_list(
"Jinja2", "Jinja2",
"k-diffusion", "k-diffusion",
"librosa", "librosa",
"omegaconf",
"parameterized", "parameterized",
"pytest", "pytest",
"pytest-timeout", "pytest-timeout",
......
src/diffusers/dependency_versions_table.py
View file @ 58c6f9cb
...@@ -20,6 +20,7 @@ deps = { ...@@ -20,6 +20,7 @@ deps = {
"k-diffusion": "k-diffusion>=0.0.12", "k-diffusion": "k-diffusion>=0.0.12",
"librosa": "librosa", "librosa": "librosa",
"numpy": "numpy", "numpy": "numpy",
"omegaconf": "omegaconf",
"parameterized": "parameterized", "parameterized": "parameterized",
"protobuf": "protobuf>=3.20.3,<4", "protobuf": "protobuf>=3.20.3,<4",
"pytest": "pytest", "pytest": "pytest",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment