mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
150 lines
4.0 KiB
Python
150 lines
4.0 KiB
Python
import os
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from omegaconf import OmegaConf
|
|
from pathlib import Path
|
|
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_rootdir(tmp_path: Path, monkeypatch: Any) -> None:
|
|
"""This may be overkill since the current tests don't need the root dir to exist"""
|
|
monkeypatch.setenv("INVOKEAI_ROOT", str(tmp_path))
|
|
|
|
|
|
init1 = OmegaConf.create(
|
|
"""
|
|
InvokeAI:
|
|
Features:
|
|
always_use_cpu: false
|
|
Memory/Performance:
|
|
max_cache_size: 5
|
|
tiled_decode: false
|
|
"""
|
|
)
|
|
|
|
init2 = OmegaConf.create(
|
|
"""
|
|
InvokeAI:
|
|
Features:
|
|
always_use_cpu: true
|
|
Memory/Performance:
|
|
max_cache_size: 2
|
|
tiled_decode: true
|
|
"""
|
|
)
|
|
|
|
init3 = OmegaConf.create(
|
|
"""
|
|
InvokeAI:
|
|
Generation:
|
|
sequential_guidance: true
|
|
attention_type: xformers
|
|
attention_slice_size: 7
|
|
forced_tiled_decode: True
|
|
Device:
|
|
device: cpu
|
|
Model Cache:
|
|
ram: 1.25
|
|
"""
|
|
)
|
|
|
|
|
|
def test_use_init(patch_rootdir):
|
|
# note that we explicitly set omegaconf dict and argv here
|
|
# so that the values aren't read from ~invokeai/invokeai.yaml and
|
|
# sys.argv respectively.
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
conf1 = InvokeAIAppConfig.get_config()
|
|
assert conf1
|
|
conf1.parse_args(conf=init1, argv=[])
|
|
assert not conf1.tiled_decode
|
|
assert conf1.max_cache_size == 5
|
|
assert not conf1.always_use_cpu
|
|
|
|
conf2 = InvokeAIAppConfig.get_config()
|
|
assert conf2
|
|
conf2.parse_args(conf=init2, argv=[])
|
|
assert conf2.tiled_decode
|
|
assert conf2.max_cache_size == 2
|
|
assert not hasattr(conf2, "invalid_attribute")
|
|
|
|
|
|
def test_legacy():
|
|
conf = InvokeAIAppConfig.get_config()
|
|
assert conf
|
|
conf.parse_args(conf=init3, argv=[])
|
|
assert conf.xformers_enabled
|
|
assert conf.device == "cpu"
|
|
assert conf.use_cpu
|
|
assert conf.ram == 1.25
|
|
assert conf.ram_cache_size == 1.25
|
|
|
|
|
|
def test_argv_override():
|
|
conf = InvokeAIAppConfig.get_config()
|
|
conf.parse_args(conf=init1, argv=["--always_use_cpu", "--max_cache=10"])
|
|
assert conf.always_use_cpu
|
|
assert conf.max_cache_size == 10
|
|
assert conf.outdir == Path("outputs") # this is the default
|
|
|
|
|
|
def test_env_override(patch_rootdir):
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
# argv overrides
|
|
conf = InvokeAIAppConfig()
|
|
conf.parse_args(conf=init1, argv=["--max_cache=10"])
|
|
assert conf.always_use_cpu is False
|
|
os.environ["INVOKEAI_always_use_cpu"] = "True"
|
|
conf.parse_args(conf=init1, argv=["--max_cache=10"])
|
|
assert conf.always_use_cpu is True
|
|
|
|
# environment variables should be case insensitive
|
|
os.environ["InvokeAI_Max_Cache_Size"] = "15"
|
|
conf = InvokeAIAppConfig()
|
|
conf.parse_args(conf=init1, argv=[])
|
|
assert conf.max_cache_size == 15
|
|
|
|
conf = InvokeAIAppConfig()
|
|
conf.parse_args(conf=init1, argv=["--no-always_use_cpu", "--max_cache=10"])
|
|
assert conf.always_use_cpu is False
|
|
assert conf.max_cache_size == 10
|
|
|
|
conf = InvokeAIAppConfig.get_config(max_cache_size=20)
|
|
conf.parse_args(conf=init1, argv=[])
|
|
assert conf.max_cache_size == 20
|
|
|
|
|
|
def test_root_resists_cwd(patch_rootdir):
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
previous = os.environ["INVOKEAI_ROOT"]
|
|
cwd = Path(os.getcwd()).resolve()
|
|
|
|
os.environ["INVOKEAI_ROOT"] = "."
|
|
conf = InvokeAIAppConfig.get_config()
|
|
conf.parse_args([])
|
|
assert conf.root_path == cwd
|
|
|
|
os.chdir("..")
|
|
assert conf.root_path == cwd
|
|
os.environ["INVOKEAI_ROOT"] = previous
|
|
os.chdir(cwd)
|
|
|
|
|
|
def test_type_coercion(patch_rootdir):
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
conf = InvokeAIAppConfig().get_config()
|
|
conf.parse_args(argv=["--root=/tmp/foobar"])
|
|
assert conf.root == Path("/tmp/foobar")
|
|
assert isinstance(conf.root, Path)
|
|
conf = InvokeAIAppConfig.get_config(root="/tmp/different")
|
|
conf.parse_args(argv=["--root=/tmp/foobar"])
|
|
assert conf.root == Path("/tmp/different")
|
|
assert isinstance(conf.root, Path)
|