Skip to content

Commit

Permalink
Add environment files.
Browse files Browse the repository at this point in the history
  • Loading branch information
robinjhuang committed Jun 26, 2024
1 parent b9aea98 commit 71b96a4
Show file tree
Hide file tree
Showing 3 changed files with 41 additions and 2 deletions.
7 changes: 5 additions & 2 deletions action.yml
Original file line number Diff line number Diff line change
Expand Up @@ -91,21 +91,24 @@ runs:
CACHE_NUMBER: 0
with:
path: ~/conda_pkgs_dir
key: ${{ runner.os }}-conda-${{ env.CACHE_NUMBER }}-${{ hashFiles('etc/environment.yml') }}
key: ${{ runner.os }}-conda-${{ env.CACHE_NUMBER }}-${{ hashFiles(format('environments/{0}-environment.yml', inputs.os)) }}

- name: '[Unix] Setup Conda'
uses: conda-incubator/setup-miniconda@v3
if: ${{ inputs.os != 'windows' }}
with:
activate-environment: comfyui
auto-activate-base: false
environment: etc/environment.yml
environment: environments/${{ inputs.os }}-environment.yml
use-only-tar-bz2: true
python-version: ${{ inputs.python_version }}

- name: '[Unix] Install Comfy-CLI'
if: ${{ inputs.os != 'windows' }}
shell: bash -el {0}
run: |
which pip
which pip3
pip3 install comfy-cli
comfy --version
comfy --skip-prompt --no-enable-telemetry env
Expand Down
11 changes: 11 additions & 0 deletions environments/linux-environment.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
name: comfyui
channels:
- pytorch
- nvidia
- defaults
dependencies:
- python>=3.9 # Specify the version of Python you need
- pytorch
- torchvision
- torchaudio
- cudatoolkit=12.1
25 changes: 25 additions & 0 deletions environments/mac-environment.yml
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit 71b96a4

Please sign in to comment.