diff --git a/hairline/service.py b/hairline/service.py index bbc6b74..f132bb8 100644 --- a/hairline/service.py +++ b/hairline/service.py @@ -50,9 +50,9 @@ _REDRAW_PROMPT = os.getenv("REDRAW_PROMPT", "填充遮罩区域的头发,皮 # 接口2 女重绘整条管线(swapHair + ComfyUI)送模型前限边。真实照片常达 1257x1495: # 全分辨率 ComfyUI 重绘要 13~21s 且激活显存把模型挤出。女性路径含 swapHair(SD WebUI ~5.3s -# 固定地板) + ComfyUI 两段串行。用 1024 优先画质(女性总耗时 ~11~12s,偶尔踩 12s); -# 若要更稳的 <12s 余量可设 REDRAW_MAX_SIDE=896(ComfyUI ~4s,女性 9~11s)。 -_REDRAW_MAX_SIDE = int(os.getenv("REDRAW_MAX_SIDE", "1024")) +# 固定地板) + ComfyUI 两段串行。1024 档画质更好但部分大图会踩 12s 线, +# 默认压到 896 兜底(ComfyUI ~4s,女性总耗时 9~11s);追画质可设 REDRAW_MAX_SIDE=1024。 +_REDRAW_MAX_SIDE = int(os.getenv("REDRAW_MAX_SIDE", "896")) def _call_local_redraw(image_png_bytes, mask_png_bytes, timeout=300.0): """直接调 ComfyUI 重绘(替代原 local_test HTTP 服务)。