Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/modules/parameters.py
View File

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

1
ldm/invoke/args.py
View File

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

2
ldm/invoke/generator/diffusers_pipeline.py
View File

@ -231,6 +231,8 @@ class StableDiffusionGeneratorPipeline(DiffusionPipeline):
run_id: str = None, **extra_step_kwargs): run_id: str = None, **extra_step_kwargs):
if run_id is None: if run_id is None:
run_id = secrets.token_urlsafe(self.ID_LENGTH) 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, yield PipelineIntermediateState(run_id=run_id, step=-1, timestep=self.scheduler.num_train_timesteps,
latents=latents) latents=latents)
# NOTE: Depends on scheduler being already initialized! # NOTE: Depends on scheduler being already initialized!