fix: add try_parse_bool for env var strings to enable config overrides of boolean values

This commit is contained in:
Manuel Schmid 2024-06-06 18:05:55 +02:00
parent 04d764820e
commit c4faf2ae6c
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
2 changed files with 12 additions and 2 deletions

View File

@ -2,13 +2,14 @@ import os
import json
import math
import numbers
import args_manager
import tempfile
import modules.flags
import modules.sdxl_styles
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
@ -209,7 +210,7 @@ def get_config_item_or_set_default(key, default_value, validator, disable_empty_
v = os.getenv(key)
if v is not None:
print(f"Environment: {key} = {v}")
config_dict[key] = v
config_dict[key] = try_parse_bool(v)
if key not in config_dict:
config_dict[key] = default_value

View File

@ -1,4 +1,6 @@
import os
from ast import literal_eval
def makedirs_with_log(path):
try:
@ -24,3 +26,10 @@ def get_files_from_folder(folder_path, extensions=None, name_filter=None):
filenames.append(path)
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