[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:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user