From c64eeda152f665febc83119714d01a9a70520e8c Mon Sep 17 00:00:00 2001
From: Robin Huang <robin.j.huang@gmail.com>
Date: Fri, 28 Jun 2024 16:49:42 -0700
Subject: [PATCH] =?UTF-8?q?Revert=20"Revert=20"Add=20integration=20test=20?=
 =?UTF-8?q?for=20Linux=20with=20Nvidia=20GPU.=20#3884=20(#3895)=E2=80=A6"?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

This reverts commit c39cf7fff09c728d573835a48106e2133c446e11.
---
 .../conda-environments/linux-environment.yml  | 28 +++++++++++
 .github/workflows/linux-integration-test.yml  | 50 +++++++++++++++++++
 2 files changed, 78 insertions(+)
 create mode 100644 .github/conda-environments/linux-environment.yml
 create mode 100644 .github/workflows/linux-integration-test.yml

diff --git a/.github/conda-environments/linux-environment.yml b/.github/conda-environments/linux-environment.yml
new file mode 100644
index 00000000000..c9ebd640616
--- /dev/null
+++ b/.github/conda-environments/linux-environment.yml
@@ -0,0 +1,28 @@
+name: comfyui
+channels:
+  - pytorch
+  - nvidia
+  - defaults
+dependencies:
+  - python>=3.9
+  - pip
+  - pytorch
+  - torchvision
+  - torchaudio
+  - pytorch-cuda=12.1
+  - pip:
+    # comfyui requirements
+    - einops
+    - transformers>=4.25.1
+    - safetensors>=0.4.2
+    - aiohttp
+    - pyyaml
+    - Pillow
+    - scipy
+    - tqdm
+    - psutil
+    # comfy-action requirements
+    - requests
+    - google-cloud-storage
+    - comfy-cli
+    - charset-normalizer
\ No newline at end of file
diff --git a/.github/workflows/linux-integration-test.yml b/.github/workflows/linux-integration-test.yml
new file mode 100644
index 00000000000..eed84a77c26
--- /dev/null
+++ b/.github/workflows/linux-integration-test.yml
@@ -0,0 +1,50 @@
+name: (Linux) ComfyUI Integration Tests
+on:
+  push:
+    branches:
+      - master
+    paths-ignore:
+      - 'app/**'
+      - 'input/**'
+      - 'output/**'
+      - 'model/**'
+      - 'notebook/**'
+      - 'script_example/**'
+      - 'tests/**'
+      - 'tests-ui/**'
+      - '.github/**'
+      - '.ci/**'
+      - 'web/**'
+  workflow_dispatch:
+  pull_request:
+    branches:
+      - master
+    paths-ignore:
+      - 'app/**'
+      - 'input/**'
+      - 'output/**'
+      - 'model/**'
+      - 'notebook/**'
+      - 'script_example/**'
+      - 'tests/**'
+      - 'tests-ui/**'
+      - '.github/**'
+      - '.ci/**'
+      - 'web/**'
+
+jobs:
+  test-workflows:
+    runs-on: [self-hosted, Linux, t4] 
+    steps:
+    - name: Test ComfyUI Workflows
+      uses: comfy-org/comfy-action@main
+      with:
+        os: linux
+        cuda_version: 12.1
+        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_filenames: "default.json"
+        gcs_bucket_name: 'comfy-ci-results'
+        google_credentials: ${{ secrets.GCS_SERVICE_ACCOUNT_JSON }}
+        output_prefix: 'ComfyUI'
+        conda_env_file: '.github/conda-environments/linux-environment.yml'
+        timeout: 50