diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index 499e9f8daf8..37c8deaefd6 100644 --- a/.github/workflows/docker-build.yml +++ b/.github/workflows/docker-build.yml @@ -60,36 +60,3 @@ jobs: slack_channel: ${{ env.CI_SLACK_CHANNEL }} title: 🤗 Results of the TRL Docker Image build slack_token: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }} - - trl-dev: - name: "Build and push TRL Dev Docker image" - runs-on: - group: aws-general-8-plus - steps: - - name: Checkout code - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1 - - - name: Set up Docker Buildx - uses: docker/setup-buildx-action@37fe631027851001ddb9b187196cc803df7f5f0e # v4.3.0 - - - name: Login to DockerHub - uses: docker/login-action@dbcb813823bdd20940b903addbd779551569679f # v4.6.0 - with: - username: ${{ secrets.DOCKERHUB_USERNAME }} - password: ${{ secrets.DOCKERHUB_PASSWORD }} - - - name: Build and Push - uses: docker/build-push-action@53b7df96c91f9c12dcc8a07bcb9ccacbed38856a # v7.3.0 - with: - context: docker/trl-dev - push: true - tags: | - huggingface/trl:dev - - - name: Post to Slack - if: always() - uses: ./.github/actions/post-slack - with: - slack_channel: ${{ env.CI_SLACK_CHANNEL }} - title: 🤗 Results of the TRL Dev Docker Image build - slack_token: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }} diff --git a/docker/trl-dev/Dockerfile b/docker/trl-dev/Dockerfile deleted file mode 100644 index 9a756a8821d..00000000000 --- a/docker/trl-dev/Dockerfile +++ /dev/null @@ -1,5 +0,0 @@ -FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel -RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/* -RUN pip install --upgrade pip uv -RUN uv pip install --system --no-cache "git+https://github.com/huggingface/trl.git#egg=trl[liger,peft,vlm]" -RUN uv pip install --system kernels liger_kernel peft trackio \ No newline at end of file