stable-diffusion-webui

1
"""this module defines internal paths used by program and is safe to import before dependencies are installed in launch.py"""
2

3
import argparse
4
import os
5
import sys
6
import shlex
7
from pathlib import Path
8

9

10
normalized_filepath = lambda filepath: str(Path(filepath).absolute())
11

12
commandline_args = os.environ.get('COMMANDLINE_ARGS', "")
13
sys.argv += shlex.split(commandline_args)
14

15
cwd = os.getcwd()
16
modules_path = os.path.dirname(os.path.realpath(__file__))
17
script_path = os.path.dirname(modules_path)
18

19
sd_configs_path = os.path.join(script_path, "configs")
20
sd_default_config = os.path.join(sd_configs_path, "v1-inference.yaml")
21
sd_model_file = os.path.join(script_path, 'model.ckpt')
22
default_sd_model_file = sd_model_file
23

24
# Parse the --data-dir flag first so we can use it as a base for our other argument default values
25
parser_pre = argparse.ArgumentParser(add_help=False)
26
parser_pre.add_argument("--data-dir", type=str, default=os.path.dirname(modules_path), help="base path where all user data is stored", )
27
cmd_opts_pre = parser_pre.parse_known_args()[0]
28

29
data_path = cmd_opts_pre.data_dir
30

31
models_path = os.path.join(data_path, "models")
32
extensions_dir = os.path.join(data_path, "extensions")
33
extensions_builtin_dir = os.path.join(script_path, "extensions-builtin")
34
config_states_dir = os.path.join(script_path, "config_states")
35
default_output_dir = os.path.join(data_path, "output")
36

37
roboto_ttf_file = os.path.join(modules_path, 'Roboto-Regular.ttf')
38