From 8dcddc47e47c639df0713bb5df8cd954ea1daff1 Mon Sep 17 00:00:00 2001
From: Robin Huang <robin.j.huang@gmail.com>
Date: Wed, 26 Jun 2024 14:35:38 -0700
Subject: [PATCH] Add env.

---
 .../conda-environments/linux-environment.yml  | 27 +++++++++++++++++++
 .../conda-environments/mac-environment.yml    | 25 +++++++++++++++++
 .github/workflows/test-workflows.yaml         |  1 +
 3 files changed, 53 insertions(+)
 create mode 100644 .github/workflows/conda-environments/linux-environment.yml
 create mode 100644 .github/workflows/conda-environments/mac-environment.yml

diff --git a/.github/workflows/conda-environments/linux-environment.yml b/.github/workflows/conda-environments/linux-environment.yml
new file mode 100644
index 00000000..e70debac
--- /dev/null
+++ b/.github/workflows/conda-environments/linux-environment.yml
@@ -0,0 +1,27 @@
+name: comfyui
+channels:
+  - pytorch
+  - nvidia
+  - defaults
+dependencies:
+  - python>=3.9  # Specify the version of Python you need
+  - pytorch
+  - torchvision
+  - torchaudio
+  - cudatoolkit=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/conda-environments/mac-environment.yml b/.github/workflows/conda-environments/mac-environment.yml
new file mode 100644
index 00000000..40ef7d61
--- /dev/null
+++ b/.github/workflows/conda-environments/mac-environment.yml
@@ -0,0 +1,25 @@
+name: comfyui
+channels:
+  - pytorch-nightly
+  - defaults
+dependencies:
+  - python>=3.9
+  - pytorch-nightly::pytorch
+  - torchvision
+  - torchaudio
+  - 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
diff --git a/.github/workflows/test-workflows.yaml b/.github/workflows/test-workflows.yaml
index e7de3eb0..1483396b 100644
--- a/.github/workflows/test-workflows.yaml
+++ b/.github/workflows/test-workflows.yaml
@@ -19,4 +19,5 @@ jobs:
         gcs_bucket_name: 'comfy-ci-results'
         google_credentials: ${{ secrets.GCS_SERVICE_ACCOUNT_JSON }}
         output_prefix: 'ComfyUI'
+        conda_env_file: 'conda-environments/linux-environment.yml'
         timeout: 50