anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[Misc] Clean logs for image gen task (#3414)

Browse Source 
Signed-off-by: wuhang <wuhang6@huawei.com>
This commit is contained in:
wuhang authored and GitHub committed 2026-05-11 12:16:05 +00:00
1 parent 3bf4f2850c
commit 3f27ffbd4d
7 files changed
+21 -19
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

vllm_omni/entrypoints/openai/api_server.py
+4 -4
View File
@@ -1606,7 +1606,7 @@ async def generate_images(request: ImageGenerationRequest, raw_request: Request)
request_id = f"img_gen-{random_uuid()}"
raw_request.state.request_metadata = RequestResponseMetadata(request_id=request_id)
logger.info(f"Generating {request.n} image(s) {size_str}")
logger.debug(f"Generating {request.n} image(s) {size_str}")
# Generate images using AsyncOmni (multi-stage mode)
result = await _generate_with_async_omni(
@@ -1626,7 +1626,7 @@ async def generate_images(request: ImageGenerationRequest, raw_request: Request)
# Extract images from result
images = _extract_images_from_result(result)
logger.info(f"Successfully generated {len(images)} image(s)")
logger.debug(f"Successfully generated {len(images)} image(s)")
# Determine output format (default to png)
output_format = _choose_output_format(request.output_format or "png", None)
@@ -1854,7 +1854,7 @@ async def edit_images(
# 4. Generate images
request_id = f"img_edit-{random_uuid()}"
raw_request.state.request_metadata = RequestResponseMetadata(request_id=request_id)
logger.info(f"Generating {n} image(s) {size_str}")
logger.debug(f"Generating {n} image(s) {size_str}")
if len(stage_configs) > 1:
# Multi-stage pipeline (e.g. GLM-Image AR+Diffusion): route through
@@ -1932,7 +1932,7 @@ async def edit_images(
)
images = _extract_images_from_result(result)
logger.info(f"Successfully generated {len(images)} image(s)")
logger.debug(f"Successfully generated {len(images)} image(s)")
# Encode images to base64
image_data = [