update workflow

ajithvcoder · Nov 4, 2024 · 6c0c26e · 6c0c26e
1 parent 0e7f972
commit 6c0c26e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ec2-pipeline.yml b/.github/workflows/ec2-pipeline.yml
@@ -146,7 +146,7 @@ jobs:
       - name: Run DVC commands in container
         run: |
           mkdir -p model_storage
-          docker run --rm --gpus=all \
+          docker run --gpus=all \
             -v model_storage:/workspace/model_storage \
             -e AWS_ACCESS_KEY_ID=${{ secrets.AWS_ACCESS_KEY_ID }} \
             -e AWS_SECRET_ACCESS_KEY=${{ secrets.AWS_SECRET_ACCESS_KEY }} \