Skip to content
This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Conversation

@marksgraham
Copy link
Collaborator

Fixes #217, where the PNDM and DDIM schedulers show different behaviour if they are used for stepping without calling set_timesteps first. @ericspod suggests objects should be immediately useable on creation, without extra methods needing to be called, so I've changed the default so the number of inference timesteps is set to the number of train timesteps during initalisation.

@Warvito Warvito self-requested a review February 10, 2023 16:26
@ericspod
Copy link
Member

Looks good!

@ericspod ericspod merged commit aef9ba2 into main Feb 10, 2023
@Warvito Warvito deleted the 217_set_default_inference_timesteps branch February 11, 2023 15:01
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Using a Scheduler Before Calling set_timestep?

3 participants