Update.

diff --git a/main.py b/main.py
index 8054509..ce86a76 100755 (executable)
--- 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