From 37ea9cc710238e031b89ba8e3483dba2e8048caf Mon Sep 17 00:00:00 2001 From: Robin Huang Date: Mon, 13 May 2024 13:07:46 -0700 Subject: [PATCH] CI/CD for testing workflows. --- .../workflows/json_examples/workflow_api.json | 107 +++++++++++++++ .../json_examples/workflow_lora_api.json | 126 ++++++++++++++++++ .github/workflows/test-workflows-windows.yaml | 24 ++++ .github/workflows/test-workflows.yaml | 47 +++++++ 4 files changed, 304 insertions(+) create mode 100644 .github/workflows/json_examples/workflow_api.json create mode 100644 .github/workflows/json_examples/workflow_lora_api.json create mode 100644 .github/workflows/test-workflows-windows.yaml create mode 100644 .github/workflows/test-workflows.yaml diff --git a/.github/workflows/json_examples/workflow_api.json b/.github/workflows/json_examples/workflow_api.json new file mode 100644 index 00000000..0a62997f --- /dev/null +++ b/.github/workflows/json_examples/workflow_api.json @@ -0,0 +1,107 @@ +{ + "3": { + "inputs": { + "seed": 156680208700286, + "steps": 20, + "cfg": 8, + "sampler_name": "euler", + "scheduler": "normal", + "denoise": 1, + "model": [ + "4", + 0 + ], + "positive": [ + "6", + 0 + ], + "negative": [ + "7", + 0 + ], + "latent_image": [ + "5", + 0 + ] + }, + "class_type": "KSampler", + "_meta": { + "title": "KSampler" + } + }, + "4": { + "inputs": { + "ckpt_name": "v1-5-pruned-emaonly.ckpt" + }, + "class_type": "CheckpointLoaderSimple", + "_meta": { + "title": "Load Checkpoint" + } + }, + "5": { + "inputs": { + "width": 512, + "height": 512, + "batch_size": 1 + }, + "class_type": "EmptyLatentImage", + "_meta": { + "title": "Empty Latent Image" + } + }, + "6": { + "inputs": { + "text": "beautiful scenery nature glass bottle landscape, , purple galaxy bottle,", + "clip": [ + "4", + 1 + ] + }, + "class_type": "CLIPTextEncode", + "_meta": { + "title": "CLIP Text Encode (Prompt)" + } + }, + "7": { + "inputs": { + "text": "text, watermark", + "clip": [ + "4", + 1 + ] + }, + "class_type": "CLIPTextEncode", + "_meta": { + "title": "CLIP Text Encode (Prompt)" + } + }, + "8": { + "inputs": { + "samples": [ + "3", + 0 + ], + "vae": [ + "4", + 2 + ] + }, + "class_type": "VAEDecode", + "_meta": { + "title": "VAE Decode" + } + }, + "9": { + "inputs": { + "filename_prefix": "ComfyUI", + "images": [ + "8", + 0 + ] + }, + "class_type": "SaveImage", + "_meta": { + "title": "Save Image" + } + } +} \ No newline at end of file diff --git a/.github/workflows/json_examples/workflow_lora_api.json b/.github/workflows/json_examples/workflow_lora_api.json new file mode 100644 index 00000000..5fa14069 --- /dev/null +++ b/.github/workflows/json_examples/workflow_lora_api.json @@ -0,0 +1,126 @@ +{ + "3": { + "inputs": { + "seed": 851616030078638, + "steps": 20, + "cfg": 8, + "sampler_name": "euler", + "scheduler": "normal", + "denoise": 1, + "model": [ + "10", + 0 + ], + "positive": [ + "6", + 0 + ], + "negative": [ + "7", + 0 + ], + "latent_image": [ + "5", + 0 + ] + }, + "class_type": "KSampler", + "_meta": { + "title": "KSampler" + } + }, + "4": { + "inputs": { + "ckpt_name": "v1-5-pruned-emaonly.ckpt" + }, + "class_type": "CheckpointLoaderSimple", + "_meta": { + "title": "Load Checkpoint" + } + }, + "5": { + "inputs": { + "width": 512, + "height": 512, + "batch_size": 1 + }, + "class_type": "EmptyLatentImage", + "_meta": { + "title": "Empty Latent Image" + } + }, + "6": { + "inputs": { + "text": "masterpiece best quality girl", + "clip": [ + "10", + 1 + ] + }, + "class_type": "CLIPTextEncode", + "_meta": { + "title": "CLIP Text Encode (Prompt)" + } + }, + "7": { + "inputs": { + "text": "bad hands", + "clip": [ + "10", + 1 + ] + }, + "class_type": "CLIPTextEncode", + "_meta": { + "title": "CLIP Text Encode (Prompt)" + } + }, + "8": { + "inputs": { + "samples": [ + "3", + 0 + ], + "vae": [ + "4", + 2 + ] + }, + "class_type": "VAEDecode", + "_meta": { + "title": "VAE Decode" + } + }, + "9": { + "inputs": { + "filename_prefix": "ComfyUI", + "images": [ + "8", + 0 + ] + }, + "class_type": "SaveImage", + "_meta": { + "title": "Save Image" + } + }, + "10": { + "inputs": { + "lora_name": "epiNoiseoffset_v2.safetensors", + "strength_model": 1, + "strength_clip": 1, + "model": [ + "4", + 0 + ], + "clip": [ + "4", + 1 + ] + }, + "class_type": "LoraLoader", + "_meta": { + "title": "Load LoRA" + } + } + } \ No newline at end of file diff --git a/.github/workflows/test-workflows-windows.yaml b/.github/workflows/test-workflows-windows.yaml new file mode 100644 index 00000000..f407fc3a --- /dev/null +++ b/.github/workflows/test-workflows-windows.yaml @@ -0,0 +1,24 @@ +name: Test ComfyUI Windows with Default Workflow +on: + push: + branches: + - master + workflow_dispatch: + + +jobs: + test-action: + runs-on: win + steps: + - name: Use ComfyUI GitHub Action + uses: drip-art/comfy-action@main + with: + os: windows + cuda_version: 12.1 + workflow_name: "default-workflow" + models-json: '{"v1-5-pruned-emaonly.ckpt": {"url": "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors?download=true", "directory": "checkpoints"}}' + workflow_json_path: '.github/workflows/json_examples/workflow_api.json' + gcs_bucket_name: 'comfy-ci-results' + google_credentials: ${{ secrets.GCS_SERVICE_ACCOUNT_JSON }} + output_prefix: 'ComfyUI' + timeout: 50 \ No newline at end of file diff --git a/.github/workflows/test-workflows.yaml b/.github/workflows/test-workflows.yaml new file mode 100644 index 00000000..c625bf84 --- /dev/null +++ b/.github/workflows/test-workflows.yaml @@ -0,0 +1,47 @@ +name: Test ComfyUI Example Workflows +on: + push: + branches: + - master + workflow_dispatch: + + + +jobs: + test-default-workflow: + strategy: + matrix: + os: [macos, linux] + include: + - os: macos + runner_label: [self-hosted, m2] + - os: linux + cuda_version: 12.1 + runner_label: [self-hosted, Linux, t4] + runs-on: ${{ matrix.runner_label }} + steps: + - name: Test Default Workflow + uses: drip-art/comfy-action@main + with: + os: ${{ matrix.os }} + cuda_version: ${{ matrix.cuda_version }} + models-json: '{"v1-5-pruned-emaonly.ckpt": {"url": "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt", "directory": "checkpoints"}}' + workflow_name: "default-workflow" + workflow_json_path: '.github/workflows/json_examples/workflow_api.json' + gcs_bucket_name: 'comfy-ci-results' + google_credentials: ${{ secrets.GCS_SERVICE_ACCOUNT_JSON }} + output_prefix: 'ComfyUI' + timeout: 50 + + - name: Test Lora Workflow + uses: drip-art/comfy-action@main + with: + os: ${{ matrix.os }} + cuda_version: ${{ matrix.cuda_version }} + models-json: '{"v1-5-pruned-emaonly.ckpt": {"url": "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt", "directory": "checkpoints"}, "epiNoiseoffset_v2.safetensors": {"url": "https://huggingface.co/adhikjoshi/epi_noiseoffset/resolve/main/epiNoiseoffset_v2.safetensors", "directory": "loras"}}' + workflow_name: "lora-workflow" + workflow_json_path: '.github/workflows/json_examples/workflow_lora_api.json' + gcs_bucket_name: 'comfy-ci-results' + google_credentials: ${{ secrets.GCS_SERVICE_ACCOUNT_JSON }} + output_prefix: 'ComfyUI' + timeout: 50 \ No newline at end of file