Update flow_matching/solver/discrete_solver.py

flow_matching/solver/discrete_solver.py

@@ -324,7 +324,7 @@ class MixtureDiscreteEulerSolver(Solver):
                 raise ImportError(
                     "tqdm is required for verbose mode. Please install it."
                 )
-            ctx = tqdm(total=n_steps, desc=f"NFE", dynamic_ncols=True, leave=True, bar_format="{desc}: {percentage:3.0f}%|{bar}| {n_fmt}/{total_fmt}{postfix}")
+            ctx = tqdm(total=n_steps, desc=f"", dynamic_ncols=True, leave=True, bar_format="{percentage:3.0f}%|{bar}| {n_fmt}/{total_fmt}{postfix}")
         else:
             ctx = nullcontext()
 
@@ -386,7 +386,7 @@ class MixtureDiscreteEulerSolver(Solver):
                 for i, objective in enumerate(args.objectives):
                     postfix[objective] = scores[i]
                 
-                ctx.set_description(f"NFE: {steps_counter}", refresh=False)
+                # ctx.set_description(f"NFE: {steps_counter}", refresh=False)
                 ctx.set_postfix({k: f"{v:.3f}" for k, v in postfix.items()}, refresh=False)
                 ctx.update(1)