[Misc] Fix error log for the diffusion stage timeout (#915)

Signed-off-by: samithuang <285365963@qq.com>
Signed-off-by: Samit <285365963@qq.com>
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
Co-authored-by: Jiangyun Zhu <riverclouds.zhu@qq.com>
This commit is contained in:
authored and GitHub committed 2026-01-23 15:16:40 +08:00
1 parent 34c9d8f2a2
commit 5d9918fa67
1 file changed
+2 -1
+2 -1
View File
@@ -345,6 +345,7 @@ class OmniBase:
)
suggestions = [
f"Ignore this warning if the model weight download / load from disk time is longer than {timeout}s.",
"Verify GPU/device assignment in config (runtime.devices) is correct.",
"Check GPU/host memory availability; reduce model or batch size if needed.",
"Check model weights path and network reachability (if loading remotely).",
@@ -353,7 +354,7 @@ class OmniBase:
formatted_suggestions = "\n".join(f" {i + 1}) {msg}" for i, msg in enumerate(suggestions))
logger.error(f"[{self._name}] Stage initialization failed. Troubleshooting Steps:\n{formatted_suggestions}")
logger.warning(f"[{self._name}] Stage initialization timeout. Troubleshooting Steps:\n{formatted_suggestions}")
def start_profile(self, stages: list[int] | None = None) -> None:
"""Start profiling for specified stages.