[diffusion] fix: fix serving with dit-layerwise-offload enabled (#16066)

Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>
This commit is contained in:
Mick
2025-12-30 00:21:42 +08:00
committed by GitHub
co-authored by Xiaoyu Zhang
parent c31f62722c
commit 8e08207c18
8 changed files with 110 additions and 9 deletions
@@ -750,7 +750,7 @@ async def benchmark(args):
)
)
print("\n" + "=" * 60)
print("=" * 60)
if args.output_file:
with open(args.output_file, "w") as f: