From 00c025bb8a725c14c6688c7f262866adcd6779e3 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Tue, 17 Sep 2024 09:45:56 +0200 Subject: [PATCH] Update. --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 8054509..ce86a76 100755 --- a/main.py +++ b/main.py @@ -488,7 +488,7 @@ def generate(model, nb, local_device=main_device): all_masks.split(args.physical_batch_size), ), dynamic_ncols=True, - desc="predict", + desc="generate", total=all_input.size(0) // args.physical_batch_size, ): changed = True -- 2.39.5