Update __init__.py

2025-08-31 18:06:20 +03:00
parent 81dac31735
commit 4870b8f081
1 changed files with 2 additions and 1 deletions
@@ -137,7 +137,8 @@ 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}")
+
    if hasattr(sample_scheduler, 'timesteps'):
        sample_scheduler.timesteps = timesteps