[schedulers] create `self.sigmas` during __init__ by yiyixuxu · Pull Request #6006 · huggingface/diffusers · GitHub
Skip to content

[schedulers] create self.sigmas during __init__ - #6006

Merged
yiyixuxu merged 7 commits into
mainfrom
scheduler-sigmas
Dec 1, 2023
Merged

yiyixuxu merged 7 commits into
mainfrom
scheduler-sigmas

Conversation

@yiyixuxu

@yiyixuxu yiyixuxu commented Nov 30, 2023 •

Copy link
Copy Markdown
Collaborator

fix this #5981

we refactored add_noise method to calculate based on sigmas in #5158, but self.sigmas is only created when you run set_timesteps()

this will cause error when the add_noise method is used without set_timesteps(), i.e. typical when used in training. In this PR we will create self.sigmas during __init__.

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.

2 participants