From a51d1fd2d53e4d412e44ecef6383996e931b9952 Mon Sep 17 00:00:00 2001
From: zhxiao <962528692@qq.com>
Date: Tue, 6 May 2025 21:48:23 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 flux_style_shaper_api/flux_style_shaper_api.py | 2 +-
 start_flux_api.sh                              | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

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