diffusers: let the scheduler do its scaling of the initial latents

Remove IPNDM scheduler; it is not behaving.
This commit is contained in:
Kevin Turner 2022-11-10 15:27:25 -08:00
parent d55e22981a
commit 1e98f4bafc
3 changed files with 2 additions and 2 deletions

View File

@ -13,7 +13,6 @@ SAMPLER_CHOICES = [
"k_lms",
"plms",
# diffusers:
"ipndm",
"pndm",
]

View File

@ -109,7 +109,6 @@ SAMPLER_CHOICES = [
'k_lms',
'plms',
# diffusers:
"ipndm",
"pndm",
]

View File

@ -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!