[Misc] Clean logs for image gen task (#3414)
Signed-off-by: wuhang <wuhang6@huawei.com>
This commit is contained in:
1 parent
3bf4f2850c
commit
3f27ffbd4d
7 files changed
+21
-19
No files matched your search
@@ -981,6 +981,11 @@ class OmniKVTransferManager:
|
||||
Returns:
|
||||
Tuple of (data dict, size) if successful, (None, 0) otherwise
|
||||
"""
|
||||
# Check if we should receive KV cache based on config
|
||||
if not self.config.need_recv_cache:
|
||||
logger.debug("Skip receiving KV cache for %s (need_recv_cache=False)", request_id)
|
||||
return None, 0
|
||||
|
||||
if not self.connector:
|
||||
logger.warning("No connector available for receiving KV cache")
|
||||
return None, 0
|
||||
@@ -990,11 +995,6 @@ class OmniKVTransferManager:
|
||||
logger.warning("Receive stages not configured")
|
||||
return None, 0
|
||||
|
||||
# Check if we should receive KV cache based on config
|
||||
if not self.config.need_recv_cache:
|
||||
logger.info(f"Skip receiving KV cache for {request_id} (need_recv_cache=False)")
|
||||
return None, 0
|
||||
|
||||
# Skip during warmup dummy run — no sender is available.
|
||||
if request_id == "dummy_req_id":
|
||||
logger.info("Skip receiving KV cache for dummy warmup request")
|
||||
|
||||
Reference in new issue
Block a user