diffusers
fix: use zero-safe geometric mean for sigmas_interpol in KDPM2DiscreteScheduler
#15003

Open

fix: use zero-safe geometric mean for sigmas_interpol in KDPM2DiscreteScheduler #15003

akhil9tiet wants to merge 8 commits into huggingface:main from akhil9tiet:fix/15002-kdpm2-nan-final-step

Delete src/diffusers/schedulers/scheduling_k_dpm_2_discrete.py

f00d4ed2

Upload fixed kdpm2.py

3c1ec4ba

fix: use zero-safe geometric mean for sigmas_interpol in KDPM2Discret…

a64f077e

Delete tests/schedulers/test_scheduler_kdpm2_discrete.py

bf3b0090

Upload updated test_kdpm2.py

a6d6fe15

test: add regression test for NaN sigmas_interpol with terminal zero …

7fe8b5a5

github-actions added tests

github-actions added schedulers

github-actions added fixes-issue

github-actions added size/S

Merge branch 'main' into fix/15002-kdpm2-nan-final-step

1b1388d5

Merge branch 'main' into fix/15002-kdpm2-nan-final-step

5b0ddb2b

Reviewers

No reviews

Assignees

No one assigned

Labels

tests size/S schedulers fixes-issue

Milestone

No milestone