Skip to content

Commit

Permalink
Run attention generator tests without spaCy (#709)
Browse files Browse the repository at this point in the history
* 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
  • Loading branch information
akx authored Jan 16, 2024
1 parent 284d3ef commit b1edd80
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
2 changes: 2 additions & 0 deletions .github/workflows/test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 0 additions & 4 deletions tests/prompts/test_attention_generator.py
Original file line number Diff line number Diff line change
@@ -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 "
Expand Down

0 comments on commit b1edd80

Please sign in to comment.