add missing platforms & tests

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

Hide file tree

Showing 4 changed files with 23 additions and 5 deletions.
diff --git a/cdr-t5/build.sh b/cdr-t5/build.sh
@@ -8,6 +8,6 @@ docker buildx build ${@:1} \
     --build-arg="MODEL_ID=Salesforce/codet5p-220m-py" \
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
-    --platform linux/amd64 \
+    --platform linux/arm64,linux/amd64 \
     .
 docker run --rm $IMAGE:$VERSION pytest
diff --git a/cht-llama-cpp/build.sh b/cht-llama-cpp/build.sh
@@ -10,6 +10,7 @@ docker buildx build ${@:1} \
     --tag $IMAGE:$VERSION \
     --platform linux/arm64,linux/amd64 \
     .
+docker run --rm $IMAGE:$VERSION pytest
 
 IMAGE=ghcr.io/premai-io/chat-vicuna-7b-q4-cpu
 docker buildx build ${@:1} \

diff --git a/cht-mpt/build.sh b/cht-mpt/build.sh
@@ -20,6 +20,9 @@ docker buildx build ${@:1} \
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
     .
+if test -z ${TESTS_SKIP_GPU+x}; then
+  docker run --rm --gpus all ghcr.io/premai-io/mpt-7b-gpu:$VERSION pytest
+fi
 
 IMAGE=ghcr.io/premai-io/mpt-7b-instruct-gpu
 docker buildx build ${@:1} \
@@ -28,3 +31,6 @@ docker buildx build ${@:1} \
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
     .
+if test -z ${TESTS_SKIP_GPU+x}; then
+  docker run --rm --gpus all ghcr.io/premai-io/mpt-7b-instruct-gpu:$VERSION pytest
+fi
diff --git a/dfs-diffusers/build.sh b/dfs-diffusers/build.sh
@@ -9,6 +9,9 @@ docker buildx build ${@:1} \
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
     .
+if test -z ${TESTS_SKIP_GPU+x}; then
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
+fi
 
 IMAGE=ghcr.io/premai-io/diffuser-stable-diffusion-1-5-gpu
 docker buildx build ${@:1} \
@@ -17,6 +20,9 @@ docker buildx build ${@:1} \
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
     .
+if test -z ${TESTS_SKIP_GPU+x}; then
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
+fi
 
 IMAGE=ghcr.io/premai-io/diffuser-stable-diffusion-2-gpu
 docker buildx build ${@:1} \
@@ -25,8 +31,9 @@ docker buildx build ${@:1} \
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
     .
-
-# docker system prune --all --force --volumes
+if test -z ${TESTS_SKIP_GPU+x}; then
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
+fi
 
 IMAGE=ghcr.io/premai-io/upscaler-stable-diffusion-x4-gpu
 docker buildx build ${@:1} \
@@ -35,6 +42,9 @@ docker buildx build ${@:1} \
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
     .
+if test -z ${TESTS_SKIP_GPU+x}; then
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
+fi
 
 IMAGE=ghcr.io/premai-io/upscaler-stable-diffusion-x2-latent-gpu
 docker buildx build ${@:1} \
@@ -43,5 +53,6 @@ docker buildx build ${@:1} \
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
     .
-
-# docker run --rm --gpus all ghcr.io/premai-io/diffuser-stable-diffusion-2-gpu:$VERSION pytest
+if test -z ${TESTS_SKIP_GPU+x}; then
+  docker run --rm --gpus all $IMAGE:$VERSION pytest
+fi