Skip to content

Commit 64c0609

Browse files
ci: use diffusers Docker image for diffusers job, add axolotl job
Use diffusers/diffusers-pytorch-cuda container for the diffusers test job — matches their own CI setup and has all system deps (libGL, etc.) pre-installed. Only override bnb with our continuous-release wheel. Previous bare-metal approach failed with libGL.so.1 missing during pytest collection (opencv-python needs it, and bare-metal runners don't have it). If the runner doesn't have Docker/nvidia-container-toolkit, the job will fail fast with a clear error and we'll fall back to installing opencv-python-headless instead. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent dc9c744 commit 64c0609

1 file changed

Lines changed: 6 additions & 10 deletions

File tree

.github/workflows/tests-integration-nightly.yml

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -157,25 +157,21 @@ jobs:
157157
name: Diffusers bnb tests
158158
if: github.repository == 'bitsandbytes-foundation/bitsandbytes'
159159
runs-on: bandb-aws-g6e-4xlarge-plus-use1-public-80 # L40S (matches diffusers CI)
160+
container:
161+
image: diffusers/diffusers-pytorch-cuda
162+
options: --gpus all --shm-size "16gb" --ipc host
160163
steps:
161164
- name: Show GPU information
162165
run: nvidia-smi
163166

164167
- uses: actions/checkout@v4
165168

166-
- name: Setup Python
167-
uses: actions/setup-python@v5
168-
with:
169-
python-version: ${{ env.PYTHON_VERSION }}
170-
171-
- name: Install torch + bnb (from continuous-release)
169+
- name: Install bnb (from continuous-release) over the image's version
172170
run: |
173-
pip install torch==${TORCH_VERSION} --index-url ${PYPI_INDEX}
174-
pip install "bitsandbytes[test] @ ${BNB_WHEEL_URL}"
171+
pip install --force-reinstall --no-deps ${BNB_WHEEL_URL}
175172
176-
- name: Install diffusers and clone matching tag
173+
- name: Clone diffusers matching installed version
177174
run: |
178-
pip install "diffusers[test]" transformers accelerate peft
179175
DIFFUSERS_VERSION=$(pip show diffusers | awk '/^Version:/ {print $2}')
180176
echo "Installed diffusers v${DIFFUSERS_VERSION}"
181177
git clone --depth=1 --branch "v${DIFFUSERS_VERSION}" \

0 commit comments

Comments
 (0)