681c16dd1e
move reading of COMMANDLINE_ARGS into paths_internal.py so --data-dir can be properly read
32 lines
1.3 KiB
Python
32 lines
1.3 KiB
Python
"""this module defines internal paths used by program and is safe to import before dependencies are installed in launch.py"""
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
import shlex
|
|
|
|
commandline_args = os.environ.get('COMMANDLINE_ARGS', "")
|
|
sys.argv += shlex.split(commandline_args)
|
|
|
|
modules_path = os.path.dirname(os.path.realpath(__file__))
|
|
script_path = os.path.dirname(modules_path)
|
|
|
|
sd_configs_path = os.path.join(script_path, "configs")
|
|
sd_default_config = os.path.join(sd_configs_path, "v1-inference.yaml")
|
|
sd_model_file = os.path.join(script_path, 'model.ckpt')
|
|
default_sd_model_file = sd_model_file
|
|
|
|
# Parse the --data-dir flag first so we can use it as a base for our other argument default values
|
|
parser_pre = argparse.ArgumentParser(add_help=False)
|
|
parser_pre.add_argument("--data-dir", type=str, default=os.path.dirname(modules_path), help="base path where all user data is stored", )
|
|
cmd_opts_pre = parser_pre.parse_known_args()[0]
|
|
|
|
data_path = cmd_opts_pre.data_dir
|
|
|
|
models_path = os.path.join(data_path, "models")
|
|
extensions_dir = os.path.join(data_path, "extensions")
|
|
extensions_builtin_dir = os.path.join(script_path, "extensions-builtin")
|
|
config_states_dir = os.path.join(script_path, "config_states")
|
|
|
|
roboto_ttf_file = os.path.join(modules_path, 'Roboto-Regular.ttf')
|