fix missing --rm --gpus all flags

premAI-io · Jul 24, 2023 · 13653e0 · 13653e0
1 parent 692d3fd
commit 13653e0
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 7 deletions.
diff --git a/a2t-whisper/build.sh b/a2t-whisper/build.sh
@@ -20,5 +20,5 @@ docker buildx build ${@:1} \
     --tag $IMAGE:$VERSION \
     .
 if test -z ${TESTS_SKIP_GPU+x}; then
-  docker run --rm $IMAGE:$VERSION pytest
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
 fi
diff --git a/cht-dolly-v2/build.sh b/cht-dolly-v2/build.sh
@@ -10,5 +10,5 @@ docker buildx build ${@:1} \
     --tag $IMAGE:$VERSION \
     .
 if test -z ${TESTS_SKIP_GPU+x}; then
-  docker run --gpus all $IMAGE:$VERSION pytest
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
 fi
diff --git a/cht-falcon/build.sh b/cht-falcon/build.sh
@@ -10,5 +10,5 @@ docker buildx build ${@:1} \
     --tag $IMAGE:$VERSION \
     .
 if test -z ${TESTS_SKIP_GPU+x}; then
-  docker run --gpus all $IMAGE:$VERSION pytest
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
 fi
diff --git a/cht-gorilla/build.sh b/cht-gorilla/build.sh
@@ -10,7 +10,7 @@ docker buildx build ${@:1} \
     --tag $IMAGE:$VERSION \
     .
 if test -z ${TESTS_SKIP_GPU+x}; then
-  docker run --gpus all $IMAGE:$VERSION pytest
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
 fi
 
 IMAGE=ghcr.io/premai-io/chat-gorilla-mpt-7b-gpu
@@ -21,5 +21,5 @@ docker buildx build ${@:1} \
     --tag $IMAGE:$VERSION \
     .
 if test -z ${TESTS_SKIP_GPU+x}; then
-  docker run --gpus all $IMAGE:$VERSION pytest
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
 fi
diff --git a/cht-mpt/build.sh b/cht-mpt/build.sh
@@ -10,7 +10,7 @@ docker buildx build ${@:1} \
     --tag $IMAGE:$VERSION \
     .
 if test -z ${TESTS_SKIP_GPU+x}; then
-  docker run --gpus all $IMAGE:$VERSION pytest
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
 fi
 
 IMAGE=ghcr.io/premai-io/mpt-7b-gpu

diff --git a/cht-xgen/build.sh b/cht-xgen/build.sh
@@ -10,5 +10,5 @@ docker buildx build ${@:1} \
     --tag $IMAGE:$VERSION \
     .
 if test -z ${TESTS_SKIP_GPU+x}; then
-  docker run --gpus all $IMAGE:$VERSION pytest
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
 fi