Skip to content

Commit 1f83920

Browse files
committed
diffusers: let the scheduler do its scaling of the initial latents
Remove IPNDM scheduler; it is not behaving.
1 parent 08c62d7 commit 1f83920

File tree

3 files changed

+2
-2
lines changed

3 files changed

+2
-2
lines changed

backend/modules/parameters.py

-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
"k_lms",
1212
"plms",
1313
# diffusers:
14-
"ipndm",
1514
"pndm",
1615
]
1716

ldm/invoke/args.py

-1
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,6 @@
105105
'k_lms',
106106
'plms',
107107
# diffusers:
108-
"ipndm",
109108
"pndm",
110109
]
111110

ldm/invoke/generator/diffusers_pipeline.py

+2
Original file line numberDiff line numberDiff line change
@@ -231,6 +231,8 @@ def generate_from_embeddings(self, latents: torch.Tensor, text_embeddings: torch
231231
run_id: str = None, **extra_step_kwargs):
232232
if run_id is None:
233233
run_id = secrets.token_urlsafe(self.ID_LENGTH)
234+
# scale the initial noise by the standard deviation required by the scheduler
235+
latents *= self.scheduler.init_noise_sigma
234236
yield PipelineIntermediateState(run_id=run_id, step=-1, timestep=self.scheduler.num_train_timesteps,
235237
latents=latents)
236238
# NOTE: Depends on scheduler being already initialized!

0 commit comments

Comments
 (0)