This commit is contained in:
zhxiao 2025-05-06 21:48:23 +08:00
parent 636a09359d
commit a51d1fd2d5
2 changed files with 3 additions and 3 deletions

View File

@ -668,7 +668,7 @@ async def fill_image(
# 处理参考图像(如果有)
reference_file_path = None
if style_image and style_image.filename:
if style_image:
reference_file_path = save_upload_file_tmp(style_image)
background_tasks.add_task(cleanup_temp_file, reference_file_path)

View File

@ -5,9 +5,9 @@ VENV_PATH="$(pwd)/venv"
# 配置参数
BASE_PORT=8001 # 起始端口号
GPU_COUNT=2 # 默认使用的GPU数量
GPU_COUNT=3 # 默认使用的GPU数量
SCRIPT_PATH="flux_style_shaper_api/flux_style_shaper_api.py" # 服务脚本路径
GPU_IDS=(3 4) # 指定要使用的GPU ID
GPU_IDS=(1 2 3) # 指定要使用的GPU ID
# 处理命令行参数
while [[ $# -gt 0 ]]; do