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 toopull/708/head
parent
284d3ef38c
commit
b1edd80487
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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 "
|
||||
|
|
|
|||
Loading…
Reference in New Issue