diff --git a/.github/workflows/build_docker_images.yml b/.github/workflows/build_docker_images.yml index f143f6de3e16..1b8e68687706 100644 --- a/.github/workflows/build_docker_images.yml +++ b/.github/workflows/build_docker_images.yml @@ -51,7 +51,6 @@ jobs: ALLOWED_IMAGES=( diffusers-pytorch-cpu diffusers-pytorch-cuda - diffusers-pytorch-xformers-cuda diffusers-pytorch-minimum-cuda diffusers-doc-builder diffusers-cli-cuda @@ -100,7 +99,6 @@ jobs: image-name: - diffusers-pytorch-cpu - diffusers-pytorch-cuda - - diffusers-pytorch-xformers-cuda - diffusers-pytorch-minimum-cuda - diffusers-doc-builder - diffusers-cli-cuda diff --git a/.github/workflows/push_tests.yml b/.github/workflows/push_tests.yml index d93ca5dbc2ee..875ce4e410c2 100644 --- a/.github/workflows/push_tests.yml +++ b/.github/workflows/push_tests.yml @@ -203,7 +203,7 @@ jobs: - name: Environment run: | diffusers-cli env - - name: Run example tests on GPU + - name: Run torch compile tests on GPU env: HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }} run: | @@ -219,50 +219,6 @@ jobs: name: torch_compile_test_reports path: reports - run_xformers_tests: - permissions: - contents: read - name: PyTorch xformers CUDA tests - - runs-on: - group: aws-g4dn-2xlarge - - container: - image: diffusers/diffusers-pytorch-xformers-cuda - options: --gpus all --shm-size "16gb" --ipc host - - steps: - - name: Checkout diffusers - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1 - with: - fetch-depth: 2 - - - name: NVIDIA-SMI - run: | - nvidia-smi - - name: Install dependencies - run: | - printf 'torch==2.10.0\ntorchvision==0.25.0\ntorchaudio==2.10.0\n' > "$UV_OVERRIDE" - uv pip install -e ".[quality,training]" - - name: Environment - run: | - diffusers-cli env - - name: Run example tests on GPU - env: - HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }} - run: | - pytest -n 1 --max-worker-restart=0 --dist=loadfile -k "xformers" --make-reports=tests_torch_xformers_cuda tests/ - - name: Failure short reports - if: ${{ failure() }} - run: cat reports/tests_torch_xformers_cuda_failures_short.txt - - - name: Test suite reports artifacts - if: ${{ always() }} - uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7.0.1 - with: - name: torch_xformers_test_reports - path: reports - run_examples_tests: permissions: contents: read diff --git a/.github/workflows/release_tests_fast.yml b/.github/workflows/release_tests_fast.yml index 83aaf4879022..3fb34c52663e 100644 --- a/.github/workflows/release_tests_fast.yml +++ b/.github/workflows/release_tests_fast.yml @@ -277,51 +277,6 @@ jobs: name: torch_compile_test_reports path: reports - run_xformers_tests: - permissions: - contents: read - name: PyTorch xformers CUDA tests - - runs-on: - group: aws-g4dn-2xlarge - - container: - image: diffusers/diffusers-pytorch-xformers-cuda - options: --gpus all --shm-size "16gb" --ipc host - - steps: - - name: Checkout diffusers - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1 - with: - fetch-depth: 2 - - - name: NVIDIA-SMI - run: | - nvidia-smi - - name: Install dependencies - run: | - printf 'torch==2.10.0\ntorchvision==0.25.0\ntorchaudio==2.10.0\n' > "$UV_OVERRIDE" - uv pip install -e ".[quality,training]" - uv pip uninstall transformers huggingface_hub && UV_PRERELEASE=allow uv pip install -U transformers@git+https://github.com/huggingface/transformers.git - - name: Environment - run: | - diffusers-cli env - - name: Run example tests on GPU - env: - HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }} - run: | - pytest -n 1 --max-worker-restart=0 --dist=loadfile -k "xformers" --make-reports=tests_torch_xformers_cuda tests/ - - name: Failure short reports - if: ${{ failure() }} - run: cat reports/tests_torch_xformers_cuda_failures_short.txt - - - name: Test suite reports artifacts - if: ${{ always() }} - uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7.0.1 - with: - name: torch_xformers_test_reports - path: reports - run_examples_tests: permissions: contents: read diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile deleted file mode 100644 index dc26eb2bf8cf..000000000000 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ /dev/null @@ -1,58 +0,0 @@ -FROM nvidia/cuda:12.9.1-runtime-ubuntu24.04 -LABEL maintainer="Hugging Face" -LABEL repository="diffusers" - -ARG PYTHON_VERSION=3.11 -ENV DEBIAN_FRONTEND=noninteractive - -RUN apt-get -y update \ - && apt-get install -y software-properties-common \ - && add-apt-repository ppa:deadsnakes/ppa && \ - apt-get update - -RUN apt install -y bash \ - build-essential \ - git \ - git-lfs \ - curl \ - ca-certificates \ - libglib2.0-0 \ - libsndfile1-dev \ - libgl1 \ - python3 \ - python3-pip \ - && apt-get clean \ - && rm -rf /var/lib/apt/lists/* - -RUN curl -LsSf https://astral.sh/uv/install.sh | sh -ENV PATH="/root/.local/bin:$PATH" -ENV VIRTUAL_ENV="/opt/venv" -ENV UV_PYTHON_INSTALL_DIR=/opt/uv/python -RUN uv venv --python ${PYTHON_VERSION} --seed ${VIRTUAL_ENV} -ENV PATH="$VIRTUAL_ENV/bin:$PATH" - -# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) -# Pin torch, torchvision, and torchaudio to a matching set so the torchvision C++ -# extension's ABI lines up with torch (otherwise torchvision::nms fails to register). -RUN uv pip install --no-cache-dir \ - torch==2.10.0 \ - torchvision==0.25.0 \ - torchaudio==2.10.0 \ - --index-url https://download.pytorch.org/whl/cu129 - -# Install compatible versions of numba/llvmlite for Python 3.10+ -RUN uv pip install --no-cache-dir \ - "llvmlite>=0.40.0" \ - "numba>=0.57.0" - -RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers[test]" - -# Extra dependencies -RUN uv pip install --no-cache-dir \ - accelerate \ - numpy==1.26.4 \ - pytorch-lightning \ - hf_xet \ - xformers - -CMD ["/bin/bash"]