CI: prefer diffusers-1.4 because it no longer requires a token

The RunwayML models still do.
This commit is contained in:
Kevin Turner 2022-11-25 14:24:00 -08:00
parent 09728dd1e0
commit ea073980b0
3 changed files with 7 additions and 11 deletions

View File

@ -14,7 +14,7 @@ jobs:
strategy:
matrix:
stable-diffusion-model:
- 'diffusers-1.5'
- diffusers-1.4
environment-yaml:
- environment-lin-amd.yml
- environment-lin-cuda.yml
@ -61,10 +61,6 @@ jobs:
path: ~/.cache/huggingface
key: ${{ env.cache-name }}
- name: Check model availability
if: steps.cache-sd-model.outputs.cache-hit != true && env.HAVE_SECRETS != 'true'
run: echo -e '\a ⛔ GitHub model cache not found, and no HUGGINGFACE_TOKEN is available. Will not be able to load Stable Diffusion.' ; exit 1
- name: Use cached conda packages
id: use-cached-conda-packages
uses: actions/cache@v3

View File

@ -14,7 +14,7 @@ jobs:
strategy:
matrix:
stable-diffusion-model:
- diffusers-1.5
- diffusers-1.4
requirements-file:
- requirements-lin-cuda.txt
- requirements-lin-amd.txt
@ -60,10 +60,6 @@ jobs:
path: ~/.cache/huggingface
key: ${{ env.cache-name }}
- name: Check model availability
if: steps.cache-sd-model.outputs.cache-hit != true && env.HAVE_SECRETS != 'true'
run: echo -e '\a ⛔ GitHub model cache not found, and no HUGGINGFACE_TOKEN is available. Will not be able to load Stable Diffusion.' ; exit 1
- name: set test prompt to main branch validation
if: ${{ github.ref == 'refs/heads/main' }}
run: echo "TEST_PROMPTS=tests/preflight_prompts.txt" >> $GITHUB_ENV

View File

@ -5,11 +5,15 @@
# model requires a model config file, a weights file,
# and the width and height of the images it
# was trained on.
diffusers-1.4:
description: Diffusers version of Stable Diffusion version 1.4
format: diffusers
repo_name: CompVis/stable-diffusion-v1-4
default: true
diffusers-1.5:
description: Diffusers version of Stable Diffusion version 1.5
format: diffusers
repo_name: runwayml/stable-diffusion-v1-5
default: true
stable-diffusion-1.5:
description: The newest Stable Diffusion version 1.5 weight file (4.27 GB)
weights: models/ldm/stable-diffusion-v1/v1-5-pruned-emaonly.ckpt