39 lines
1.6 KiB
Python
Executable File
39 lines
1.6 KiB
Python
Executable File
"""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
|
|
from pathlib import Path
|
|
|
|
|
|
normalized_filepath = lambda filepath: str(Path(filepath).absolute())
|
|
|
|
commandline_args = os.environ.get('COMMANDLINE_ARGS', "")
|
|
sys.argv += shlex.split(commandline_args)
|
|
|
|
cwd = os.getcwd()
|
|
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", )
|
|
parser_pre.add_argument("--models-dir", type=str, default=None, help="base path where models are stored; overrides --data-dir", )
|
|
cmd_opts_pre = parser_pre.parse_known_args()[0]
|
|
|
|
data_path = cmd_opts_pre.data_dir
|
|
|
|
models_path = cmd_opts_pre.models_dir if cmd_opts_pre.models_dir else 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")
|
|
default_output_dir = os.path.join(data_path, "outputs")
|
|
|
|
roboto_ttf_file = os.path.join(modules_path, 'Roboto-Regular.ttf')
|