mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
diffusers: let the scheduler do its scaling of the initial latents
Remove IPNDM scheduler; it is not behaving.
This commit is contained in:
parent
d55e22981a
commit
1e98f4bafc
@ -13,7 +13,6 @@ SAMPLER_CHOICES = [
|
||||
"k_lms",
|
||||
"plms",
|
||||
# diffusers:
|
||||
"ipndm",
|
||||
"pndm",
|
||||
]
|
||||
|
||||
|
@ -109,7 +109,6 @@ SAMPLER_CHOICES = [
|
||||
'k_lms',
|
||||
'plms',
|
||||
# diffusers:
|
||||
"ipndm",
|
||||
"pndm",
|
||||
]
|
||||
|
||||
|
@ -231,6 +231,8 @@ class StableDiffusionGeneratorPipeline(DiffusionPipeline):
|
||||
run_id: str = None, **extra_step_kwargs):
|
||||
if run_id is None:
|
||||
run_id = secrets.token_urlsafe(self.ID_LENGTH)
|
||||
# scale the initial noise by the standard deviation required by the scheduler
|
||||
latents *= self.scheduler.init_noise_sigma
|
||||
yield PipelineIntermediateState(run_id=run_id, step=-1, timestep=self.scheduler.num_train_timesteps,
|
||||
latents=latents)
|
||||
# NOTE: Depends on scheduler being already initialized!
|
||||
|
Loading…
Reference in New Issue
Block a user