From 56b1295e0526be9445a66895688d2258b7c121f2 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Wed, 29 May 2024 22:19:11 +0200 Subject: [PATCH] build --- .../build-nightly-ci-docker-images.yml | 20 ------------------- .github/workflows/self-scheduled.yml | 4 ++-- 2 files changed, 2 insertions(+), 22 deletions(-) diff --git a/.github/workflows/build-nightly-ci-docker-images.yml b/.github/workflows/build-nightly-ci-docker-images.yml index 691369c765aec7..0b1b7df5f8a2ed 100644 --- a/.github/workflows/build-nightly-ci-docker-images.yml +++ b/.github/workflows/build-nightly-ci-docker-images.yml @@ -15,16 +15,6 @@ jobs: name: "Nightly PyTorch + Stable TensorFlow" runs-on: [intel-cpu, 8-cpu, ci] steps: - - name: Cleanup disk - run: | - sudo ls -l /usr/local/lib/ - sudo ls -l /usr/share/ - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo rm -rf /usr/local/lib/android - sudo rm -rf /usr/share/dotnet - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - name: Set up Docker Buildx uses: docker/setup-buildx-action@v2 @@ -52,16 +42,6 @@ jobs: name: "Nightly PyTorch + DeepSpeed" runs-on: [intel-cpu, 8-cpu, ci] steps: - - name: Cleanup disk - run: | - sudo ls -l /usr/local/lib/ - sudo ls -l /usr/share/ - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo rm -rf /usr/local/lib/android - sudo rm -rf /usr/share/dotnet - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - name: Set up Docker Buildx uses: docker/setup-buildx-action@v2 diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index cf5887fc9aee03..27375662fbbe1b 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -267,8 +267,8 @@ jobs: machine_type: [single-gpu, multi-gpu] runs-on: ['${{ matrix.machine_type }}', nvidia-gpu, t4, '${{ inputs.runner }}'] container: -# image: huggingface/transformers-pytorch-deepspeed-latest-gpu - image: ${{ inputs.docker }} + image: huggingface/transformers-pytorch-deepspeed-latest-gpu +# image: ${{ inputs.docker }} options: --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ steps: - name: Update clone