[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
@@ -212,7 +212,7 @@ class DiffusionModelRunner(OmniConnectorModelRunnerMixin):
|
||||
peak_allocated_gb = peak_allocated_bytes / (1024**3)
|
||||
pool_overhead_gb = peak_reserved_gb - peak_allocated_gb
|
||||
|
||||
logger.info(
|
||||
logger.debug(
|
||||
"Peak GPU memory (this request): %.2f GB reserved, %.2f GB allocated, %.2f GB pool overhead (%.1f%%)",
|
||||
peak_reserved_gb,
|
||||
peak_allocated_gb,
|
||||
|
||||
Reference in new issue
Block a user