feat: add tests.json

runpod-workers · Oct 30, 2023 · b2d3c58 · b2d3c58
1 parent 58469ca
commit b2d3c58
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/.github/workflows/tests.json b/.github/workflows/tests.json
@@ -0,0 +1,16 @@
+[
+    {
+        "hardwareConfig": {
+            "endpointConfig": {
+                "gpuIds": "ADA_24,AMPERE_24,AMPERE_48,AMPERE_80"
+            }
+        },
+        "input": {
+            "dataset_url": "https://github.com/runpod-workers/sample-inputs/raw/main/images/runpod_dog2.zip",
+            "instance_prompt": "a photo of sks dog",
+            "batch_size": 32,
+            "training_steps": 1,
+            "local_directory": "./cat"
+        }
+    }
+]
diff --git a/README.md b/README.md
@@ -4,7 +4,7 @@
 
 [![CI | Test Handler](https://github.com/runpod-workers/worker-sdxl-tuner/actions/workflows/CI-test_handler.yml/badge.svg)](https://github.com/runpod-workers/worker-sdxl-tuner/actions/workflows/CI-test_handler.yml)
 &nbsp;
-[![Docker Image](https://github.com/runpod-workers/worker-template/actions/workflows/CD-docker_dev.yml/badge.svg)](https://github.com/runpod-workers/worker-template/actions/workflows/CD-docker_dev.yml)
+[![CD | Build-Test-Release](https://github.com/runpod-workers/worker-sdxl-tuner/actions/workflows/build-test-release.yml/badge.svg)](https://github.com/runpod-workers/worker-sdxl-tuner/actions/workflows/build-test-release.yml)
 
 🚀 | This serverless worker is used for fine tuning SDXL. It supports dynamic auto-scaling using the built-in RunPod autoscaling feature.