feature: add video progress bar

10 months ago · 3b2265f82d
parent b5a0e65f35
commit 3b2265f82d
1 changed files with 5 additions and 5 deletions
--- a/imaginairy/modules/sgm/diffusionmodules/sampling.py
+++ b/imaginairy/modules/sgm/diffusionmodules/sampling.py
@ -70,11 +70,11 @@ class BaseDiffusionSampler:
            print(f"Sampler: {self.__class__.__name__}")
            print(f"Discretization: {self.discretization.__class__.__name__}")
            print(f"Guider: {self.guider.__class__.__name__}")
-            sigma_generator = tqdm(
-                sigma_generator,
-                total=num_sigmas,
-                desc=f"Sampling with {self.__class__.__name__} for {num_sigmas} steps",
-            )
+        sigma_generator = tqdm(
+            sigma_generator,
+            total=num_sigmas,
+            desc=f"{self.__class__.__name__}",
+        )
        return sigma_generator