zredxiao
/
KICCO_AI_IMAGE
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
flux_style_shaper_api/flux_style_shaper_api.py
View File

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

4
start_flux_api.sh
View File

@ -5,9 +5,9 @@ VENV_PATH="$(pwd)/venv"
# 配置参数 # 配置参数
BASE_PORT=8001 # 起始端口号 BASE_PORT=8001 # 起始端口号
GPU_COUNT=2 # 默认使用的GPU数量 GPU_COUNT=3 # 默认使用的GPU数量
SCRIPT_PATH="flux_style_shaper_api/flux_style_shaper_api.py" # 服务脚本路径 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 while [[ $# -gt 0 ]]; do