diff --git a/flux_style_shaper_api/flux_style_shaper_api.py b/flux_style_shaper_api/flux_style_shaper_api.py index 12e41c2a..c31fa4e6 100644 --- a/flux_style_shaper_api/flux_style_shaper_api.py +++ b/flux_style_shaper_api/flux_style_shaper_api.py @@ -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) diff --git a/start_flux_api.sh b/start_flux_api.sh index a3776a3d..0648ecb4 100644 --- a/start_flux_api.sh +++ b/start_flux_api.sh @@ -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