kl_optimal_scheduler divided torch.arange(n) by (n - 1), which is a division by zero when n == 1, making the first sigma NaN and corrupting the whole generation. steps=1 is a valid KSampler/BasicScheduler input. Clamp the divisor to at least 1 so a single step starts at sigma_max. |
||
|---|---|---|
| .. | ||
| folder_path_test.py | ||
| kl_optimal_scheduler_test.py | ||
| model_detection_test.py | ||
| seedvr_vae_forward_test.py | ||
| test_seedvr2_dtype.py | ||
| test_seedvr2_internals.py | ||
| test_seedvr2_model.py | ||
| test_seedvr2_vae_decode.py | ||
| test_seedvr2_vae_tiled.py | ||