2023-03-25 21:05:12 +08:00
""" this module defines internal paths used by program and is safe to import before dependencies are installed in launch.py """
import argparse
import os
script_path = os . path . dirname ( os . path . dirname ( os . path . realpath ( __file__ ) ) )
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 ( os . path . dirname ( os . path . realpath ( __file__ ) ) ) , 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 " )
2023-03-30 05:46:03 +08:00
config_states_dir = os . path . join ( script_path , " config_states " )