feature: add video progress bar

This commit is contained in:
Bryce 2023-11-24 12:33:36 -08:00
parent b5a0e65f35
commit 3b2265f82d

View File

@ -73,7 +73,7 @@ class BaseDiffusionSampler:
sigma_generator = tqdm(
sigma_generator,
total=num_sigmas,
desc=f"Sampling with {self.__class__.__name__} for {num_sigmas} steps",
desc=f"{self.__class__.__name__}",
)
return sigma_generator