Run attention generator tests without spaCy (#709)

* CI: use CPU wheel repo for PyTorch

* Run attention generator tests without spaCy

dynamicprompts v0.26.0+ doesn't require spaCy for attention, so we can test it in CI here too
pull/708/head
Aarni Koskela 2024-01-16 11:11:49 +02:00 committed by GitHub
parent 284d3ef38c
commit b1edd80487
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 4 deletions

View File

@ -23,6 +23,8 @@ jobs:
with:
python-version: "3.10"
cache: "pip"
- name: Install CPU Torch
run: pip3 install torch torchaudio --index-url https://download.pytorch.org/whl/cpu
- name: Install dependencies
run: |
python -m pip install pytest-cov

View File

@ -1,11 +1,7 @@
import pytest
from sd_dynamic_prompts.attention_generator import SpecialSyntaxAwareAttentionGenerator
@pytest.mark.slow
def test_default_generator():
pytest.importorskip("spacy")
generator = SpecialSyntaxAwareAttentionGenerator()
for prompt in generator.generate(
"purple cat singing opera, artistic, painting "