diff --git a/.github/workflows/pr_modular_tests.yml b/.github/workflows/pr_modular_tests.yml index 89b502d364ec..fab180e0fe1f 100644 --- a/.github/workflows/pr_modular_tests.yml +++ b/.github/workflows/pr_modular_tests.yml @@ -117,7 +117,7 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install -e ".[quality,test]" #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1 uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps diff --git a/.github/workflows/pr_tests.yml b/.github/workflows/pr_tests.yml index 52c894bee05d..3f6b8b4cd04f 100644 --- a/.github/workflows/pr_tests.yml +++ b/.github/workflows/pr_tests.yml @@ -114,7 +114,7 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install -e ".[quality,test]" #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1 uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps @@ -191,7 +191,7 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install -e ".[quality,test]" - name: Environment run: | @@ -242,7 +242,7 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install -e ".[quality,test]" # TODO (sayakpaul, DN6): revisit `--no-deps` uv pip install -U peft@git+https://github.com/huggingface/peft.git --no-deps uv pip install -U tokenizers diff --git a/setup.py b/setup.py index 8f1d266df6de..45bffd376351 100644 --- a/setup.py +++ b/setup.py @@ -101,6 +101,7 @@ "datasets", "filelock", "flax>=0.4.1", + "ftfy", "hf-doc-builder>=0.3.0", "httpx<1.0.0", "huggingface-hub>=0.34.0,<2.0", @@ -221,12 +222,14 @@ def run(self): extras["training"] = deps_list("accelerate", "datasets", "protobuf", "tensorboard", "Jinja2", "peft", "timm") extras["test"] = deps_list( "compel", + "ftfy", "GitPython", "datasets", "Jinja2", "invisible-watermark", "librosa", "parameterized", + "protobuf", "pytest", "pytest-timeout", "pytest-xdist", @@ -235,6 +238,7 @@ def run(self): "sentencepiece", "scipy", "tiktoken", + "torchsde", "torchvision", "transformers", "phonemizer", diff --git a/src/diffusers/dependency_versions_table.py b/src/diffusers/dependency_versions_table.py index 87b6a5bd3827..5b3f84213a2e 100644 --- a/src/diffusers/dependency_versions_table.py +++ b/src/diffusers/dependency_versions_table.py @@ -8,6 +8,7 @@ "datasets": "datasets", "filelock": "filelock", "flax": "flax>=0.4.1", + "ftfy": "ftfy", "hf-doc-builder": "hf-doc-builder>=0.3.0", "httpx": "httpx<1.0.0", "huggingface-hub": "huggingface-hub>=0.34.0,<2.0",