Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

make invokeai.yaml more hierarchical; fix list configuration bug

Browse Source
This commit is contained in:
Lincoln Stein
2023-05-17 12:19:19 -04:00
parent eadfd239a8
commit b7c5a39685
3 changed files with 143 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_config.py
View File

@ -10,17 +10,21 @@ from invokeai.app.invocations.generate import TextToImageInvocation
init1 = OmegaConf.create(
'''
globals:
nsfw_checker: False
max_loaded_models: 5
InvokeAI:
Features:
nsfw_checker: False
Memory/Performance:
max_loaded_models: 5
'''
)
init2 = OmegaConf.create(
'''
globals:
nsfw_checker: True
max_loaded_models: 2
InvokeAI:
Features:
nsfw_checker: true
Memory/Performance:
max_loaded_models: 2
'''
)
@ -50,7 +54,7 @@ def test_env_override():
conf = InvokeAIAppConfig(conf=init1,argv=['--max_loaded=10'])
assert conf.nsfw_checker==False
os.environ['INVOKEAI_globals_nsfw_checker'] = 'True'
os.environ['INVOKEAI_nsfw_checker'] = 'True'
conf = InvokeAIAppConfig(conf=init1,argv=['--max_loaded=10'])
assert conf.nsfw_checker==True