fix: add try_parse_bool for env var strings to enable config overrides of boolean values
This commit is contained in:
parent
04d764820e
commit
c4faf2ae6c
|
|
@ -2,13 +2,14 @@ import os
|
||||||
import json
|
import json
|
||||||
import math
|
import math
|
||||||
import numbers
|
import numbers
|
||||||
|
|
||||||
import args_manager
|
import args_manager
|
||||||
import tempfile
|
import tempfile
|
||||||
import modules.flags
|
import modules.flags
|
||||||
import modules.sdxl_styles
|
import modules.sdxl_styles
|
||||||
|
|
||||||
from modules.model_loader import load_file_from_url
|
from modules.model_loader import load_file_from_url
|
||||||
from modules.extra_utils import makedirs_with_log, get_files_from_folder
|
from modules.extra_utils import makedirs_with_log, get_files_from_folder, try_parse_bool
|
||||||
from modules.flags import OutputFormat, Performance, MetadataScheme
|
from modules.flags import OutputFormat, Performance, MetadataScheme
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -209,7 +210,7 @@ def get_config_item_or_set_default(key, default_value, validator, disable_empty_
|
||||||
v = os.getenv(key)
|
v = os.getenv(key)
|
||||||
if v is not None:
|
if v is not None:
|
||||||
print(f"Environment: {key} = {v}")
|
print(f"Environment: {key} = {v}")
|
||||||
config_dict[key] = v
|
config_dict[key] = try_parse_bool(v)
|
||||||
|
|
||||||
if key not in config_dict:
|
if key not in config_dict:
|
||||||
config_dict[key] = default_value
|
config_dict[key] = default_value
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,6 @@
|
||||||
import os
|
import os
|
||||||
|
from ast import literal_eval
|
||||||
|
|
||||||
|
|
||||||
def makedirs_with_log(path):
|
def makedirs_with_log(path):
|
||||||
try:
|
try:
|
||||||
|
|
@ -24,3 +26,10 @@ def get_files_from_folder(folder_path, extensions=None, name_filter=None):
|
||||||
filenames.append(path)
|
filenames.append(path)
|
||||||
|
|
||||||
return filenames
|
return filenames
|
||||||
|
|
||||||
|
|
||||||
|
def try_parse_bool(value: str) -> str | bool:
|
||||||
|
value_eval = literal_eval(value.strip().title())
|
||||||
|
if type(value_eval) is bool:
|
||||||
|
return value_eval
|
||||||
|
return value
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue