[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

auto_round/estimation.py

@@ -166,9 +166,18 @@ def estimate_time(num_layers, iters, nsamples, batch_size):
     return total_seconds
 
 
-def dry_run_estimate(model_name, scheme_bits, group_size, model_dtype="float16",
-                     batch_size=8, seqlen=2048, nsamples=128, iters=200,
-                     trust_remote_code=True, platform="hf"):
+def dry_run_estimate(
+    model_name,
+    scheme_bits,
+    group_size,
+    model_dtype="float16",
+    batch_size=8,
+    seqlen=2048,
+    nsamples=128,
+    iters=200,
+    trust_remote_code=True,
+    platform="hf",
+):
     """Run a dry-run estimation and return a dict of estimates.
 
     Args:
@@ -249,8 +258,10 @@ def print_dry_run_report(estimates):
     print(f"  Estimated peak VRAM:    {estimates['peak_vram_str']}")
     print(f"  Estimated output size:  {estimates['output_size_str']}")
     print(f"  Estimated time:         {estimates['estimated_time_str']}")
-    print(f"    (batch_size={estimates['batch_size']}, seqlen={estimates['seqlen']}, "
-          f"nsamples={estimates['nsamples']}, iters={estimates['iters']})")
+    print(
+        f"    (batch_size={estimates['batch_size']}, seqlen={estimates['seqlen']}, "
+        f"nsamples={estimates['nsamples']}, iters={estimates['iters']})"
+    )
     print(border)
     print("  NOTE: These are rough estimates. Actual values depend on")
     print("  hardware, model architecture, and runtime conditions.")