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 排除， 由网盘单独上传。
102 lines
2.6 KiB
Python
102 lines
2.6 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# @File : setup.py
|
|
# @Time : 2020/1/15
|
|
# @Author : yangchaojie (yangchaojie@immomo.com)
|
|
|
|
import os
|
|
import sys
|
|
import shutil
|
|
import numpy
|
|
import tempfile
|
|
|
|
from setuptools import setup
|
|
from setuptools.extension import Extension
|
|
|
|
from Cython.Build import cythonize
|
|
from Cython.Distutils import build_ext
|
|
|
|
import platform
|
|
|
|
|
|
def get_root_path(root):
|
|
if os.path.dirname(root) in ['', '.']:
|
|
return os.path.basename(root)
|
|
else:
|
|
return get_root_path(os.path.dirname(root))
|
|
|
|
|
|
def copy_file(src, dest):
|
|
if os.path.exists(dest):
|
|
return
|
|
|
|
if not os.path.exists(os.path.dirname(dest)):
|
|
os.makedirs(os.path.dirname(dest))
|
|
if os.path.isdir(src):
|
|
shutil.copytree(src, dest)
|
|
else:
|
|
shutil.copyfile(src, dest)
|
|
|
|
|
|
def touch_init_file():
|
|
init_file_name = os.path.join(tempfile.mkdtemp(), '__init__.py')
|
|
with open(init_file_name, 'w'):
|
|
pass
|
|
return init_file_name
|
|
|
|
|
|
|
|
|
|
def compose_extensions(root='.'):
|
|
for file_ in os.listdir(root):
|
|
abs_file = os.path.join(root, file_)
|
|
|
|
if os.path.isfile(abs_file):
|
|
if abs_file.endswith('.py'):
|
|
extensions.append(Extension(get_root_path(abs_file) + '.*', [abs_file]))
|
|
elif abs_file.endswith('.c') or abs_file.endswith('.pyc'):
|
|
continue
|
|
else:
|
|
copy_file(abs_file, os.path.join(build_root_dir, abs_file))
|
|
if abs_file.endswith('__init__.py'):
|
|
copy_file(init_file, os.path.join(build_root_dir, abs_file))
|
|
|
|
else:
|
|
if os.path.basename(abs_file) in ignore_folders :
|
|
continue
|
|
if os.path.basename(abs_file) in conf_folders:
|
|
copy_file(abs_file, os.path.join(build_root_dir, abs_file))
|
|
compose_extensions(abs_file)
|
|
|
|
|
|
|
|
# if __name__ == '__main__':
|
|
build_root_dir = 'build/lib.' + platform.system().lower() + '-' + platform.machine() + '-' + str(
|
|
sys.version_info.major) + '.' + str(sys.version_info.minor)
|
|
|
|
print(build_root_dir)
|
|
|
|
extensions = []
|
|
ignore_folders = ['build', 'new_ref_zao_color_0818', 'ref_hair_online_0703_local', 'ref_分好类别', '.git']
|
|
conf_folders = ['conf']
|
|
|
|
|
|
init_file = touch_init_file()
|
|
print(init_file)
|
|
|
|
|
|
compose_extensions()
|
|
os.remove(init_file)
|
|
|
|
setup(
|
|
name='moxie_hairstyle',
|
|
version='1.0',
|
|
ext_modules=cythonize(
|
|
extensions,
|
|
nthreads=16,
|
|
compiler_directives=dict(always_allow_keywords=True),
|
|
include_path=[numpy.get_include()]),
|
|
cmdclass=dict(build_ext=build_ext))
|
|
|
|
# python setup.py build_ext |