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 排除， 由网盘单独上传。
37 lines
1.5 KiB
Python
Executable File
37 lines
1.5 KiB
Python
Executable File
import importlib
|
|
|
|
|
|
always_true_func = lambda *args, **kwargs: True
|
|
|
|
|
|
class CondFunc:
|
|
def __new__(cls, orig_func, sub_func, cond_func=always_true_func):
|
|
self = super(CondFunc, cls).__new__(cls)
|
|
if isinstance(orig_func, str):
|
|
func_path = orig_func.split('.')
|
|
for i in range(len(func_path)-1, -1, -1):
|
|
try:
|
|
resolved_obj = importlib.import_module('.'.join(func_path[:i]))
|
|
break
|
|
except ImportError:
|
|
pass
|
|
try:
|
|
for attr_name in func_path[i:-1]:
|
|
resolved_obj = getattr(resolved_obj, attr_name)
|
|
orig_func = getattr(resolved_obj, func_path[-1])
|
|
setattr(resolved_obj, func_path[-1], lambda *args, **kwargs: self(*args, **kwargs))
|
|
except AttributeError:
|
|
print(f"Warning: Failed to resolve {orig_func} for CondFunc hijack")
|
|
pass
|
|
self.__init__(orig_func, sub_func, cond_func)
|
|
return lambda *args, **kwargs: self(*args, **kwargs)
|
|
def __init__(self, orig_func, sub_func, cond_func):
|
|
self.__orig_func = orig_func
|
|
self.__sub_func = sub_func
|
|
self.__cond_func = cond_func
|
|
def __call__(self, *args, **kwargs):
|
|
if not self.__cond_func or self.__cond_func(self.__orig_func, *args, **kwargs):
|
|
return self.__sub_func(self.__orig_func, *args, **kwargs)
|
|
else:
|
|
return self.__orig_func(*args, **kwargs)
|