Skip to content

Conversation

@sayakpaul
Copy link
Member

What does this PR do?

As discussed internally. I have only introduced the change to EDMEuler but could do it for EDMDPM too once I get an approval.

Co-authored-by: Suraj Patil <[email protected]>
@sayakpaul sayakpaul requested a review from yiyixuxu April 12, 2024 06:35
@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@donthomasitos
Copy link

Graph of the different schedules: https://www.desmos.com/calculator/p7ow4l1dl1

Copy link
Collaborator

@yiyixuxu yiyixuxu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

thanks!
I left a feedback

@sayakpaul sayakpaul requested a review from yiyixuxu April 26, 2024 14:33
@sayakpaul
Copy link
Member Author

@yiyixuxu thanks for the feedback. Just adddressed them.

Copy link
Collaborator

@yiyixuxu yiyixuxu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

looks good to me!
just need to update the docstring

@sayakpaul sayakpaul marked this pull request as ready for review April 27, 2024 01:54
@sayakpaul
Copy link
Member Author

@yiyixuxu just FYI, I have also propagated the changes to EDMDPMSolverMultistepScheduler too.

@sayakpaul sayakpaul merged commit 56bd7e6 into main Apr 27, 2024
@sayakpaul sayakpaul deleted the feat/sigma-schedule-edm branch April 27, 2024 02:10
sayakpaul added a commit that referenced this pull request Dec 23, 2024
* introduce sigma schedule.

Co-authored-by: Suraj Patil <[email protected]>

* address yiyi

* update docstrings.

* implement the schedule for EDMDPMSolverMultistepScheduler

---------

Co-authored-by: Suraj Patil <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants