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 排除， 由网盘单独上传。
61 lines
2.3 KiB
Python
Executable File
61 lines
2.3 KiB
Python
Executable File
import os
|
|
|
|
import torch
|
|
|
|
from modules import shared
|
|
from modules.shared import cmd_opts
|
|
|
|
|
|
def initialize():
|
|
"""Initializes fields inside the shared module in a controlled manner.
|
|
|
|
Should be called early because some other modules you can import mingt need these fields to be already set.
|
|
"""
|
|
|
|
os.makedirs(cmd_opts.hypernetwork_dir, exist_ok=True)
|
|
|
|
from modules import options, shared_options
|
|
shared.options_templates = shared_options.options_templates
|
|
shared.opts = options.Options(shared_options.options_templates, shared_options.restricted_opts)
|
|
shared.restricted_opts = shared_options.restricted_opts
|
|
try:
|
|
shared.opts.load(shared.config_filename)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
from modules import devices
|
|
devices.device, devices.device_interrogate, devices.device_gfpgan, devices.device_esrgan, devices.device_codeformer = \
|
|
(devices.cpu if any(y in cmd_opts.use_cpu for y in [x, 'all']) else devices.get_optimal_device() for x in ['sd', 'interrogate', 'gfpgan', 'esrgan', 'codeformer'])
|
|
|
|
devices.dtype = torch.float32 if cmd_opts.no_half else torch.float16
|
|
devices.dtype_vae = torch.float32 if cmd_opts.no_half or cmd_opts.no_half_vae else torch.float16
|
|
devices.dtype_inference = torch.float32 if cmd_opts.precision == 'full' else devices.dtype
|
|
|
|
if cmd_opts.precision == "half":
|
|
msg = "--no-half and --no-half-vae conflict with --precision half"
|
|
assert devices.dtype == torch.float16, msg
|
|
assert devices.dtype_vae == torch.float16, msg
|
|
assert devices.dtype_inference == torch.float16, msg
|
|
devices.force_fp16 = True
|
|
devices.force_model_fp16()
|
|
|
|
shared.device = devices.device
|
|
shared.weight_load_location = None if cmd_opts.lowram else "cpu"
|
|
|
|
from modules import shared_state
|
|
shared.state = shared_state.State()
|
|
|
|
from modules import styles
|
|
shared.prompt_styles = styles.StyleDatabase(shared.styles_filename)
|
|
|
|
from modules import interrogate
|
|
shared.interrogator = interrogate.InterrogateModels("interrogate")
|
|
|
|
from modules import shared_total_tqdm
|
|
shared.total_tqdm = shared_total_tqdm.TotalTQDM()
|
|
|
|
from modules import memmon, devices
|
|
shared.mem_mon = memmon.MemUsageMonitor("MemMon", devices.device, shared.opts)
|
|
shared.mem_mon.start()
|
|
|