mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-11-07 09:20:35 +00:00
95a8fa31a9
while the previous version did produce much better blending it also makes images that lack detail for some reason. tests: Added more tests to help catch this sort of thing earlies fix: found that median blur is really slow, so I made sure we only do it on downsampled masks. Was taking like 3 minutes to run on the large pearl girl picture on M1 - docs: update examples
25 lines
579 B
Python
25 lines
579 B
Python
import pytest
|
|
from click.testing import CliRunner
|
|
|
|
from imaginairy.cmds import imagine_cmd
|
|
from imaginairy.utils import get_device
|
|
from tests import TESTS_FOLDER
|
|
|
|
|
|
@pytest.mark.skipif(get_device() == "cpu", reason="Too slow to run on CPU")
|
|
def test_imagine_cmd():
|
|
runner = CliRunner()
|
|
result = runner.invoke(
|
|
imagine_cmd,
|
|
[
|
|
"gold coins",
|
|
"--steps",
|
|
"25",
|
|
"--outdir",
|
|
f"{TESTS_FOLDER}/test_output",
|
|
"--seed",
|
|
"703425280",
|
|
],
|
|
)
|
|
assert result.exit_code == 0
|