diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index dc3aa102be..0fa5b26eaa 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -38,8 +38,8 @@ jobs: run: | apt update apt install -y libpq-dev postgresql-client - uv pip install -e ".[quality]" - uv pip install -r benchmarks/requirements.txt + uv pip install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow -r benchmarks/requirements.txt - name: Environment run: | python utils/print_env.py diff --git a/.github/workflows/build_pr_documentation.yml b/.github/workflows/build_pr_documentation.yml index f47645c1f6..f45d92d604 100644 --- a/.github/workflows/build_pr_documentation.yml +++ b/.github/workflows/build_pr_documentation.yml @@ -31,7 +31,7 @@ jobs: - name: Install doc-builder run: | - uv pip install --system git+https://github.com/huggingface/doc-builder.git@main + uv pip install --prerelease=allow --system git+https://github.com/huggingface/doc-builder.git@main - name: Check documentation links run: | diff --git a/.github/workflows/nightly_tests.yml b/.github/workflows/nightly_tests.yml index 8b7e57e912..c548b0fa50 100644 --- a/.github/workflows/nightly_tests.yml +++ b/.github/workflows/nightly_tests.yml @@ -71,11 +71,11 @@ jobs: run: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git - #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 install pytest-reportlog + uv pip install --prerelease=allow -e ".[quality]" + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 + uv pip install --prerelease=allow pytest-reportlog - name: Environment run: | python utils/print_env.py @@ -125,12 +125,12 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install peft@git+https://github.com/huggingface/peft.git - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git - #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 install pytest-reportlog + uv pip install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow peft@git+https://github.com/huggingface/peft.git + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 + uv pip install --prerelease=allow pytest-reportlog - name: Environment run: python utils/print_env.py @@ -193,9 +193,9 @@ jobs: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality,training]" - #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 install --prerelease=allow -e ".[quality,training]" + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 - name: Environment run: | python utils/print_env.py @@ -235,12 +235,12 @@ jobs: run: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install peft@git+https://github.com/huggingface/peft.git - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git - #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 install pytest-reportlog + uv pip install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow peft@git+https://github.com/huggingface/peft.git + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 + uv pip install --prerelease=allow pytest-reportlog - name: Environment run: | python utils/print_env.py @@ -286,11 +286,11 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install peft@git+https://github.com/huggingface/peft.git - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git - #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 install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow peft@git+https://github.com/huggingface/peft.git + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 - name: Environment run: | @@ -362,14 +362,14 @@ jobs: run: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install -U ${{ matrix.config.backend }} + uv pip install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow -U ${{ matrix.config.backend }} if [ "${{ join(matrix.config.additional_deps, ' ') }}" != "" ]; then - uv pip install ${{ join(matrix.config.additional_deps, ' ') }} + uv pip install --prerelease=allow ${{ join(matrix.config.additional_deps, ' ') }} fi - uv pip install pytest-reportlog - #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 install --prerelease=allow pytest-reportlog + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 - name: Environment run: | python utils/print_env.py @@ -415,11 +415,11 @@ jobs: run: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install -U bitsandbytes optimum_quanto - #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 install pytest-reportlog + uv pip install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow -U bitsandbytes optimum_quanto + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 + uv pip install --prerelease=allow pytest-reportlog - name: Environment run: | python utils/print_env.py @@ -531,10 +531,10 @@ jobs: # shell: arch -arch arm64 bash {0} # run: | # ${CONDA_RUN} pip install --upgrade pip uv -# ${CONDA_RUN} uv pip install -e ".[quality]" -# ${CONDA_RUN} uv pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu -# ${CONDA_RUN} uv pip install accelerate@git+https://github.com/huggingface/accelerate -# ${CONDA_RUN} uv pip install pytest-reportlog +# ${CONDA_RUN} uv pip install --prerelease=allow -e ".[quality]" +# ${CONDA_RUN} uv pip install --prerelease=allow torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu +# ${CONDA_RUN} uv pip install --prerelease=allow accelerate@git+https://github.com/huggingface/accelerate +# ${CONDA_RUN} uv pip install --prerelease=allow pytest-reportlog # - name: Environment # shell: arch -arch arm64 bash {0} # run: | @@ -587,10 +587,10 @@ jobs: # shell: arch -arch arm64 bash {0} # run: | # ${CONDA_RUN} pip install --upgrade pip uv -# ${CONDA_RUN} uv pip install -e ".[quality]" -# ${CONDA_RUN} uv pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu -# ${CONDA_RUN} uv pip install accelerate@git+https://github.com/huggingface/accelerate -# ${CONDA_RUN} uv pip install pytest-reportlog +# ${CONDA_RUN} uv pip install --prerelease=allow -e ".[quality]" +# ${CONDA_RUN} uv pip install --prerelease=allow torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu +# ${CONDA_RUN} uv pip install --prerelease=allow accelerate@git+https://github.com/huggingface/accelerate +# ${CONDA_RUN} uv pip install --prerelease=allow pytest-reportlog # - name: Environment # shell: arch -arch arm64 bash {0} # run: | diff --git a/.github/workflows/pr_modular_tests.yml b/.github/workflows/pr_modular_tests.yml index 13c228621f..7a9b7d4944 100644 --- a/.github/workflows/pr_modular_tests.yml +++ b/.github/workflows/pr_modular_tests.yml @@ -108,10 +108,10 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" - #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 + uv pip install --prerelease=allow -e ".[quality]" + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps - name: Environment run: | diff --git a/.github/workflows/pr_test_fetcher.yml b/.github/workflows/pr_test_fetcher.yml index 83b2ab4edb..3cd0b4a0d6 100644 --- a/.github/workflows/pr_test_fetcher.yml +++ b/.github/workflows/pr_test_fetcher.yml @@ -33,7 +33,7 @@ jobs: fetch-depth: 0 - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install --prerelease=allow -e ".[quality]" - name: Environment run: | python utils/print_env.py @@ -89,8 +89,8 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install accelerate + uv pip install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow accelerate - name: Environment run: | diff --git a/.github/workflows/pr_tests.yml b/.github/workflows/pr_tests.yml index 674e62ff44..e792236e63 100644 --- a/.github/workflows/pr_tests.yml +++ b/.github/workflows/pr_tests.yml @@ -114,10 +114,10 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" - #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 + uv pip install --prerelease=allow -e ".[quality]" + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps - name: Environment run: | @@ -142,7 +142,7 @@ jobs: - name: Run example PyTorch CPU tests if: ${{ matrix.config.framework == 'pytorch_examples' }} run: | - uv pip install ".[training]" + uv pip install --prerelease=allow ".[training]" pytest -n 4 --max-worker-restart=0 --dist=loadfile \ --make-reports=tests_${{ matrix.config.report }} \ examples @@ -191,7 +191,7 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install --prerelease=allow -e ".[quality]" - name: Environment run: | @@ -242,13 +242,13 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install --prerelease=allow -e ".[quality]" # 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 - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps - #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 install --prerelease=allow -U peft@git+https://github.com/huggingface/peft.git --no-deps + uv pip install --prerelease=allow -U tokenizers + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 - name: Environment run: | diff --git a/.github/workflows/pr_tests_gpu.yml b/.github/workflows/pr_tests_gpu.yml index 468979d379..18a7cb9569 100644 --- a/.github/workflows/pr_tests_gpu.yml +++ b/.github/workflows/pr_tests_gpu.yml @@ -88,7 +88,7 @@ jobs: fetch-depth: 2 - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install --prerelease=allow -e ".[quality]" - name: Environment run: | python utils/print_env.py @@ -129,10 +129,10 @@ jobs: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git - #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 install --prerelease=allow -e ".[quality]" + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 - name: Environment run: | @@ -199,11 +199,11 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install peft@git+https://github.com/huggingface/peft.git - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git - #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 install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow peft@git+https://github.com/huggingface/peft.git + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 - name: Environment run: | @@ -264,9 +264,9 @@ jobs: nvidia-smi - name: Install dependencies run: | - #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 install -e ".[quality,training]" + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 + uv pip install --prerelease=allow -e ".[quality,training]" - name: Environment run: | @@ -276,7 +276,7 @@ jobs: env: HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }} run: | - uv pip install ".[training]" + uv pip install --prerelease=allow ".[training]" pytest -n 1 --max-worker-restart=0 --dist=loadfile --make-reports=examples_torch_cuda examples/ - name: Failure short reports diff --git a/.github/workflows/push_tests.yml b/.github/workflows/push_tests.yml index 7b1c441d3d..7bc80bb1ea 100644 --- a/.github/workflows/push_tests.yml +++ b/.github/workflows/push_tests.yml @@ -34,7 +34,7 @@ jobs: fetch-depth: 2 - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install --prerelease=allow -e ".[quality]" - name: Environment run: | python utils/print_env.py @@ -74,10 +74,10 @@ jobs: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git - #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 install --prerelease=allow -e ".[quality]" + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 - name: Environment run: | python utils/print_env.py @@ -126,11 +126,11 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install peft@git+https://github.com/huggingface/peft.git - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git - #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 install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow peft@git+https://github.com/huggingface/peft.git + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 - name: Environment run: | @@ -181,9 +181,9 @@ jobs: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality,training]" - #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 install --prerelease=allow -e ".[quality,training]" + #uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git + uv pip uninstall transformers huggingface_hub && uv pip install --prerelease=allow transformers==4.57.1 - name: Environment run: | python utils/print_env.py @@ -225,7 +225,7 @@ jobs: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality,training]" + uv pip install --prerelease=allow -e ".[quality,training]" - name: Environment run: | python utils/print_env.py @@ -265,7 +265,7 @@ jobs: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality,training]" + uv pip install --prerelease=allow -e ".[quality,training]" - name: Environment run: | @@ -275,7 +275,7 @@ jobs: env: HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }} run: | - uv pip install ".[training]" + uv pip install --prerelease=allow ".[training]" pytest -n 1 --max-worker-restart=0 --dist=loadfile --make-reports=examples_torch_cuda examples/ - name: Failure short reports diff --git a/.github/workflows/push_tests_fast.yml b/.github/workflows/push_tests_fast.yml index 38cbffaa63..49176b8122 100644 --- a/.github/workflows/push_tests_fast.yml +++ b/.github/workflows/push_tests_fast.yml @@ -60,7 +60,7 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install --prerelease=allow -e ".[quality]" - name: Environment run: | @@ -77,7 +77,7 @@ jobs: - name: Run example PyTorch CPU tests if: ${{ matrix.config.framework == 'pytorch_examples' }} run: | - uv pip install ".[training]" + uv pip install --prerelease=allow ".[training]" pytest -n 4 --max-worker-restart=0 --dist=loadfile \ --make-reports=tests_${{ matrix.config.report }} \ examples diff --git a/.github/workflows/push_tests_mps.yml b/.github/workflows/push_tests_mps.yml index 2d6feb5928..426163f7cb 100644 --- a/.github/workflows/push_tests_mps.yml +++ b/.github/workflows/push_tests_mps.yml @@ -41,10 +41,10 @@ jobs: shell: arch -arch arm64 bash {0} run: | ${CONDA_RUN} python -m pip install --upgrade pip uv - ${CONDA_RUN} python -m uv pip install -e ".[quality,test]" - ${CONDA_RUN} python -m uv pip install torch torchvision torchaudio - ${CONDA_RUN} python -m uv pip install accelerate@git+https://github.com/huggingface/accelerate.git - ${CONDA_RUN} python -m uv pip install transformers --upgrade + ${CONDA_RUN} python -m uv pip install --prerelease=allow -e ".[quality,test]" + ${CONDA_RUN} python -m uv pip install --prerelease=allow torch torchvision torchaudio + ${CONDA_RUN} python -m uv pip install --prerelease=allow accelerate@git+https://github.com/huggingface/accelerate.git + ${CONDA_RUN} python -m uv pip install --prerelease=allow transformers --upgrade - name: Environment shell: arch -arch arm64 bash {0} diff --git a/.github/workflows/release_tests_fast.yml b/.github/workflows/release_tests_fast.yml index efdd6ea2b6..fee52eb91e 100644 --- a/.github/workflows/release_tests_fast.yml +++ b/.github/workflows/release_tests_fast.yml @@ -32,7 +32,7 @@ jobs: fetch-depth: 2 - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install --prerelease=allow -e ".[quality]" - name: Environment run: | python utils/print_env.py @@ -72,8 +72,8 @@ jobs: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git + uv pip install --prerelease=allow -e ".[quality]" + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git - name: Environment run: | python utils/print_env.py @@ -122,9 +122,9 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install peft@git+https://github.com/huggingface/peft.git - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git + uv pip install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow peft@git+https://github.com/huggingface/peft.git + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git - name: Environment run: | @@ -172,9 +172,9 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" - uv pip install peft@git+https://github.com/huggingface/peft.git - uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git + uv pip install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow peft@git+https://github.com/huggingface/peft.git + uv pip uninstall accelerate && uv pip install --prerelease=allow -U accelerate@git+https://github.com/huggingface/accelerate.git - name: Environment run: | @@ -231,7 +231,7 @@ jobs: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality,training]" + uv pip install --prerelease=allow -e ".[quality,training]" - name: Environment run: | python utils/print_env.py @@ -273,7 +273,7 @@ jobs: nvidia-smi - name: Install dependencies run: | - uv pip install -e ".[quality,training]" + uv pip install --prerelease=allow -e ".[quality,training]" - name: Environment run: | python utils/print_env.py @@ -315,7 +315,7 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality,training]" + uv pip install --prerelease=allow -e ".[quality,training]" - name: Environment run: | @@ -325,7 +325,7 @@ jobs: env: HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }} run: | - uv pip install ".[training]" + uv pip install --prerelease=allow ".[training]" pytest -n 1 --max-worker-restart=0 --dist=loadfile --make-reports=examples_torch_cuda examples/ - name: Failure short reports diff --git a/.github/workflows/run_tests_from_a_pr.yml b/.github/workflows/run_tests_from_a_pr.yml index fa8c579dd7..0ae5ab2f75 100644 --- a/.github/workflows/run_tests_from_a_pr.yml +++ b/.github/workflows/run_tests_from_a_pr.yml @@ -63,8 +63,8 @@ jobs: - name: Install pytest run: | - uv pip install -e ".[quality]" - uv pip install peft + uv pip install --prerelease=allow -e ".[quality]" + uv pip install --prerelease=allow peft - name: Run tests env: diff --git a/setup.py b/setup.py index d52d37787f..48a77a89df 100644 --- a/setup.py +++ b/setup.py @@ -103,7 +103,7 @@ _deps = [ "flax>=0.4.1", "hf-doc-builder>=0.3.0", "httpx<1.0.0", - "huggingface-hub>=0.34.0,<2.0", + "huggingface-hub==v1.3.0.rc0", "requests-mock==1.10.0", "importlib_metadata", "invisible-watermark>=0.2.0", @@ -150,7 +150,7 @@ _deps = [ # this is a lookup table with items like: # -# tokenizers: "huggingface-hub==0.8.0" +# tokenizers: "huggingface-hub==v1.3.0.rc0" # packaging: "packaging" # # some of the values are versioned whereas others aren't. diff --git a/src/diffusers/dependency_versions_table.py b/src/diffusers/dependency_versions_table.py index 6e5ac630ab..b2c81b185d 100644 --- a/src/diffusers/dependency_versions_table.py +++ b/src/diffusers/dependency_versions_table.py @@ -10,7 +10,7 @@ deps = { "flax": "flax>=0.4.1", "hf-doc-builder": "hf-doc-builder>=0.3.0", "httpx": "httpx<1.0.0", - "huggingface-hub": "huggingface-hub>=0.34.0,<2.0", + "huggingface-hub": "huggingface-hub==v1.3.0.rc0", "requests-mock": "requests-mock==1.10.0", "importlib_metadata": "importlib_metadata", "invisible-watermark": "invisible-watermark>=0.2.0",