From 769d9df138f9dee0bb2ba9191e2d497bdb5b511e Mon Sep 17 00:00:00 2001
From: Robin Huang <robin.j.huang@gmail.com>
Date: Wed, 26 Jun 2024 14:35:32 -0700
Subject: [PATCH] add conda env input.

---
 action.yml                         |  5 ++++-
 environments/linux-environment.yml | 27 ---------------------------
 environments/mac-environment.yml   | 25 -------------------------
 3 files changed, 4 insertions(+), 53 deletions(-)
 delete mode 100644 environments/linux-environment.yml
 delete mode 100644 environments/mac-environment.yml

diff --git a/action.yml b/action.yml
index fe0d410..a361269 100644
--- a/action.yml
+++ b/action.yml
@@ -50,6 +50,9 @@ inputs:
     description: "Comfy Org API endpoint"
     required: false
     default: "https://api.comfy.org/upload-artifact"
+  conda_env_file:
+    description: "Conda environment file. Path is relative to your repository root."
+    required: true
 runs:
   using: "composite"
   steps:
@@ -104,7 +107,7 @@ runs:
       with:
         activate-environment: comfyui
         auto-activate-base: false
-        environment-file: ${{ github.action_path }}/environments/${{ inputs.os }}-environment.yml
+        environment-file: ${{ inputs.conda_env_file }}
         use-only-tar-bz2: true
 
     - name: '[Unix] Install Comfy-CLI'
diff --git a/environments/linux-environment.yml b/environments/linux-environment.yml
deleted file mode 100644
index e70deba..0000000
--- a/environments/linux-environment.yml
+++ /dev/null
@@ -1,27 +0,0 @@
-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/environments/mac-environment.yml b/environments/mac-environment.yml
deleted file mode 100644
index 40ef7d6..0000000
--- a/environments/mac-environment.yml
+++ /dev/null
@@ -1,25 +0,0 @@
-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