From 6739401fb73044d99f4d3d1f597f71b10cf6c6ba Mon Sep 17 00:00:00 2001 From: Charlie Marshak Date: Thu, 23 Jan 2025 11:34:44 -0800 Subject: [PATCH] remove display --- src/distmetrics/transformer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/distmetrics/transformer.py b/src/distmetrics/transformer.py index 194a6fe..a44061d 100644 --- a/src/distmetrics/transformer.py +++ b/src/distmetrics/transformer.py @@ -299,7 +299,6 @@ def _estimate_logit_params_via_streamed_patches( mininterval=2, disable=(not tqdm_enabled), dynamic_ncols=True, - display=True, ): chip_mean, chip_logvar = model(patch_batch) for k, (sy, sx) in enumerate(slices): @@ -407,7 +406,6 @@ def _estimate_logit_params_via_folding( mininterval=2, disable=(not tqdm_enabled), dynamic_ncols=True, - display=True, ): # change last dimension from P**2 to P, P; use -1 because won't always have batch_size as 0th dimension batch_s = slice(batch_size * i, batch_size * (i + 1))