mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
114 lines
3.8 KiB
YAML
114 lines
3.8 KiB
YAML
name: Test Invoke with Conda
|
|
on:
|
|
push:
|
|
branches:
|
|
- 'main'
|
|
- 'development'
|
|
- 'use-bearer-token-to-download-model-debug'
|
|
pull_request_target:
|
|
branches:
|
|
- 'main'
|
|
- 'development'
|
|
|
|
jobs:
|
|
os_matrix:
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
stable-diffusion-model: ['https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt']
|
|
os: [ubuntu-latest, macos-latest]
|
|
include:
|
|
- os: ubuntu-latest
|
|
environment-file: environment.yml
|
|
default-shell: bash -l {0}
|
|
- os: macos-latest
|
|
environment-file: environment-mac.yml
|
|
default-shell: bash -l {0}
|
|
name: Test invoke.py on ${{ matrix.os }} with ${{ matrix.stable-diffusion-model }}
|
|
runs-on: ${{ matrix.os }}
|
|
defaults:
|
|
run:
|
|
shell: ${{ matrix.default-shell }}
|
|
steps:
|
|
- name: Checkout sources
|
|
uses: actions/checkout@v3
|
|
|
|
- name: setup miniconda
|
|
uses: conda-incubator/setup-miniconda@v2
|
|
with:
|
|
auto-activate-base: false
|
|
auto-update-conda: false
|
|
miniconda-version: latest
|
|
|
|
- name: set test prompt to main branch validation
|
|
if: ${{ github.ref == 'refs/heads/main' }}
|
|
run: echo "TEST_PROMPTS=tests/preflight_prompts.txt" >> $GITHUB_ENV
|
|
|
|
- name: set test prompt to development branch validation
|
|
if: ${{ github.ref == 'refs/heads/development' }}
|
|
run: echo "TEST_PROMPTS=tests/dev_prompts.txt" >> $GITHUB_ENV
|
|
|
|
- name: set test prompt to Pull Request validation
|
|
if: ${{ github.ref != 'refs/heads/main' && github.ref != 'refs/heads/development' }}
|
|
run: echo "TEST_PROMPTS=tests/validate_pr_prompt.txt" >> $GITHUB_ENV
|
|
|
|
- name: set conda environment name
|
|
run: echo "CONDA_ENV_NAME=invokeai" >> $GITHUB_ENV
|
|
|
|
- name: Use Cached Stable Diffusion Model
|
|
id: cache-stable-diffusion-model
|
|
uses: actions/cache@v3
|
|
env:
|
|
cache-name: cache-sd-${{ matrix.stable-diffusion-model }}
|
|
with:
|
|
path: models/ldm/stable-diffusion-v1/model.ckpt
|
|
key: ${{ env.cache-name }}
|
|
restore-keys: ${{ env.cache-name }}
|
|
|
|
- name: Download Stable Diffusion v1.4 Model
|
|
if: ${{ steps.cache-sd-v1-4.outputs.cache-hit != 'true' }}
|
|
run: |
|
|
[[ -d models/ldm/stable-diffusion-v1 ]] \
|
|
|| mkdir -p models/ldm/stable-diffusion-v1
|
|
[[ -r models/ldm/stable-diffusion-v1/model.ckpt ]] \
|
|
|| curl \
|
|
-H "Authorization: Bearer ${{ secrets.HUGGINGFACE_TOKEN }}" \
|
|
-o models/ldm/stable-diffusion-v1/model.ckpt \
|
|
-L ${{ matrix.stable-diffusion-model }}
|
|
|
|
- name: Activate Conda Env
|
|
uses: conda-incubator/setup-miniconda@v2
|
|
with:
|
|
activate-environment: ${{ env.CONDA_ENV_NAME }}
|
|
environment-file: ${{ matrix.environment-file }}
|
|
|
|
- name: Use Cached Huggingface and Torch models
|
|
id: cache-hugginface-torch
|
|
uses: actions/cache@v3
|
|
env:
|
|
cache-name: cache-hugginface-torch
|
|
with:
|
|
path: ~/.cache
|
|
key: ${{ env.cache-name }}
|
|
restore-keys: |
|
|
${{ env.cache-name }}-${{ hashFiles('scripts/preload_models.py') }}
|
|
|
|
- name: run preload_models.py
|
|
run: python scripts/preload_models.py
|
|
|
|
- name: Run the tests
|
|
run: |
|
|
time python scripts/invoke.py \
|
|
--from_file ${{ env.TEST_PROMPTS }}
|
|
|
|
- name: export conda env
|
|
run: |
|
|
mkdir -p outputs/img-samples
|
|
conda env export --name ${{ env.CONDA_ENV_NAME }} > outputs/img-samples/environment-${{ runner.os }}.yml
|
|
|
|
- name: Archive results
|
|
uses: actions/upload-artifact@v3
|
|
with:
|
|
name: results_${{ matrix.os }}
|
|
path: outputs/img-samples
|