colomi
0eb61f3e60
包含： - hair_service_sd: 换发型/换发色算法服务 (端口 8801) - photo_service: LoRA 训练调度服务 (端口 32678) - stable-diffusion-webui: SD WebUI 推理服务 (端口 57860) - kohya_ss_home: 训练环境代码 - meidaojia: 监控测试脚本 - setup.sh: 一键部署脚本 (conda环境恢复 + 配置生成 + 完整性检查) - start_all_services.sh: 启动3个服务 - configure.ini.template: 路径模板化 (BASE_DIR自动推导) - conda_envs/py310.yml: py310 环境定义 大文件 (weights/, models/, data/, conda_envs/*.tar.gz 等) 通过 .gitignore 排除， 由网盘单独上传。
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')
|