Update __init__.py

2025-08-31 18:09:24 +03:00
parent 4870b8f081
commit cedac626e2
1 changed files with 2 additions and 0 deletions
@@ -112,6 +112,7 @@ def get_scheduler(scheduler, steps, start_step, end_step, shift, device, transfo
    end_idx = len(timesteps) - 1

    if log_timesteps:
+        log.info(f"----------- Scheduler info -------")
        log.info(f"Total timesteps: {timesteps}")

    if isinstance(start_step, float):
@@ -138,6 +139,7 @@ def get_scheduler(scheduler, steps, start_step, end_step, shift, device, transfo
    if log_timesteps:
        log.info(f"Using timesteps: {timesteps}")
        log.info(f"Using sigmas: {sample_scheduler.sigmas}")
+        log.info(f"---------------------------")

    if hasattr(sample_scheduler, 'timesteps'):
        sample_scheduler.timesteps = timesteps