ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-21 23:09:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added timestep modifications to lcm scheduler for more evenly spaced timesteps

Browse Source
This commit is contained in:
Jaret Burkett
2023-11-17 23:26:52 -07:00
parent 6280284d8b
commit fbec68681d
4 changed files with 277 additions and 175 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
toolkit/stable_diffusion_model.py
View File

@@ -836,8 +836,9 @@ class StableDiffusion:
bleed_latents: torch.FloatTensor = None,
**kwargs,
):
timesteps_to_run = self.noise_scheduler.timesteps[start_timesteps:total_timesteps]
for timestep in tqdm(self.noise_scheduler.timesteps[start_timesteps:total_timesteps], leave=False):
for timestep in tqdm(timesteps_to_run, leave=False):
noise_pred = self.predict_noise(
latents,
text_embeddings,