From 6d2b4013f8b24f3042a9c7704e824dfadb0f9271 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Thu, 21 Sep 2023 12:37:27 -0400 Subject: [PATCH 1/2] Respect INVOKEAI_ prefix on environment variables --- invokeai/app/services/config/invokeai_config.py | 1 + 1 file changed, 1 insertion(+) diff --git a/invokeai/app/services/config/invokeai_config.py b/invokeai/app/services/config/invokeai_config.py index 65bf9b9eba..51ccf45704 100644 --- a/invokeai/app/services/config/invokeai_config.py +++ b/invokeai/app/services/config/invokeai_config.py @@ -277,6 +277,7 @@ class InvokeAIAppConfig(InvokeAISettings): class Config: validate_assignment = True + env_prefix = "INVOKEAI" def parse_args(self, argv: Optional[list[str]] = None, conf: Optional[DictConfig] = None, clobber=False): """ From 28a1a6939f9e99deea367be8204ddbe6b62494bd Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Thu, 21 Sep 2023 12:43:34 -0400 Subject: [PATCH 2/2] add regression test --- tests/test_config.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/test_config.py b/tests/test_config.py index a950a9c06f..2b2492f6a6 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -121,6 +121,12 @@ def test_env_override(patch_rootdir): conf.parse_args(conf=init1, argv=[]) assert conf.max_cache_size == 20 + # make sure that prefix is respected + del os.environ["INVOKEAI_always_use_cpu"] + os.environ["always_use_cpu"] = "True" + conf.parse_args(conf=init1, argv=[]) + assert conf.always_use_cpu is False + def test_root_resists_cwd(patch_rootdir): from invokeai.app.services.config import InvokeAIAppConfig