Skip to content

Commit a82ca6d

Browse files
CI Remove bnb docker image build from GH workflow (#2859)
In #2858, the bitsandbytes tests and Dockerfiles were removed from the PEFT repo. However, I forgot to update the GitHub workflow that builds these images. This PR now fixes that. Also changed the image used by the torch.compile workflow, as it was using a bnb image.
1 parent 5311e87 commit a82ca6d

File tree

2 files changed

+2
-69
lines changed

2 files changed

+2
-69
lines changed

.github/workflows/build_docker_images.yml

Lines changed: 0 additions & 67 deletions
Original file line numberDiff line numberDiff line change
@@ -81,70 +81,3 @@ jobs:
8181
title: 🤗 Results of the PEFT-GPU docker build
8282
status: ${{ job.status }}
8383
slack_token: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }}
84-
85-
latest-cuda-bnb-source:
86-
name: "Latest Peft GPU + bnb source [dev]"
87-
runs-on:
88-
group: aws-general-8-plus
89-
steps:
90-
- name: Set up Docker Buildx
91-
uses: docker/setup-buildx-action@b5ca514318bd6ebac0fb2aedd5d36ec1b5c232a2 # v3.10.0
92-
- name: Check out code
93-
uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8 # v5.0.0
94-
with:
95-
persist-credentials: false
96-
- name: Login to DockerHub
97-
uses: docker/login-action@74a5d142397b4f367a81961eba4e8cd7edddf772 # v3.4.0
98-
with:
99-
username: ${{ secrets.DOCKERHUB_USERNAME }}
100-
password: ${{ secrets.DOCKERHUB_PASSWORD }}
101-
102-
- name: Build and Push GPU
103-
uses: docker/build-push-action@14487ce63c7a62a4a324b0bfb37086795e31c6c1 # v6.16.0
104-
with:
105-
context: ./docker/peft-gpu-bnb-source
106-
push: true
107-
tags: huggingface/peft-gpu-bnb-source
108-
109-
- name: Post to Slack
110-
if: always()
111-
uses: huggingface/hf-workflows/.github/actions/post-slack@3f88d63d3761558a32e8e46fc2a8536e04bb2aea # main from Feb 2025-02-24
112-
with:
113-
slack_channel: ${{ env.CI_SLACK_CHANNEL }}
114-
title: 🤗 Results of the PEFT-GPU (bnb source / HF latest) docker build
115-
status: ${{ job.status }}
116-
slack_token: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }}
117-
118-
latest-cuda-bnb-source-latest:
119-
name: "Latest Peft GPU + bnb source [accelerate / peft / transformers latest]"
120-
runs-on:
121-
group: aws-general-8-plus
122-
steps:
123-
- name: Set up Docker Buildx
124-
uses: docker/setup-buildx-action@b5ca514318bd6ebac0fb2aedd5d36ec1b5c232a2 # v3.10.0
125-
- name: Check out code
126-
uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8 # v5.0.0
127-
with:
128-
persist-credentials: false
129-
- name: Login to DockerHub
130-
uses: docker/login-action@74a5d142397b4f367a81961eba4e8cd7edddf772 # v3.4.0
131-
with:
132-
username: ${{ secrets.DOCKERHUB_USERNAME }}
133-
password: ${{ secrets.DOCKERHUB_PASSWORD }}
134-
135-
- name: Build and Push GPU
136-
uses: docker/build-push-action@14487ce63c7a62a4a324b0bfb37086795e31c6c1 # v6.16.0
137-
with:
138-
context: ./docker/peft-gpu-bnb-latest
139-
push: true
140-
tags: huggingface/peft-gpu-bnb-latest
141-
142-
- name: Post to Slack
143-
if: always()
144-
uses: huggingface/hf-workflows/.github/actions/post-slack@3f88d63d3761558a32e8e46fc2a8536e04bb2aea # main from Feb 2025-02-24
145-
with:
146-
slack_channel: ${{ env.CI_SLACK_CHANNEL }}
147-
title: 🤗 Results of the PEFT-GPU (bnb source / HF source) docker build
148-
status: ${{ job.status }}
149-
slack_token: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }}
150-

.github/workflows/torch_compile_tests.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,10 @@ jobs:
2626
env:
2727
PEFT_DEBUG_WITH_TORCH_COMPILE: 1
2828
CUDA_VISIBLE_DEVICES: "0"
29-
TEST_TYPE: "single_gpu_huggingface/peft-gpu-bnb-latest:latest"
29+
TEST_TYPE: "single_gpu_huggingface/peft-gpu:latest"
3030
USE_PYTORCH_NIGHTLY: "${{ github.event.inputs.pytorch_nightly }}"
3131
container:
32-
image: "huggingface/peft-gpu-bnb-latest:latest"
32+
image: "huggingface/peft-gpu:latest"
3333
options: --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
3434
defaults:
3535
run:

0 commit comments

Comments
 (0)