Skip to content

use the PLMS schedule when PNDM has fewer than four steps - #15109

Open
chessing234 wants to merge 1 commit into
huggingface:mainfrom
chessing234:fix/pndm-few-steps
Open

chessing234 wants to merge 1 commit into
huggingface:mainfrom
chessing234:fix/pndm-few-steps

Conversation

@chessing234

Copy link
Copy Markdown

fixes #14953

the PRK tile is order 4, so set_timesteps(1), 2, or 3 crashed on a numpy broadcast. those step counts now use the PLMS schedule. 4 or more steps are unchanged.

@yiyixuxu

Made with Cursor

the order-4 PRK tile cannot be built for 1, 2, or 3 steps, and set_timesteps crashed.

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

PNDMScheduler.set_timesteps crashes with NumPy broadcasting error for fewer than 4 inference steps

1 participant