From 68518a3194f2f0710f7460f14e7a8c1f8065cb56 Mon Sep 17 00:00:00 2001 From: madroid Date: Sun, 13 Oct 2024 21:47:33 +0800 Subject: [PATCH] FLUX: add generate images tqdm desc --- flux/flux/flux.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flux/flux/flux.py b/flux/flux/flux.py index c9f23f7d..3fd044ac 100644 --- a/flux/flux/flux.py +++ b/flux/flux/flux.py @@ -184,7 +184,7 @@ class FluxPipeline: mx.eval(x_t) images = [] - for i in tqdm(range(len(x_t)), disable=not progress): + for i in tqdm(range(len(x_t)), disable=not progress, desc="generate images"): images.append(self.decode(x_t[i : i + 1])) mx.eval(images[-1]) images = mx.concatenate(images, axis=0)