diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-20 12:29:58 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-20 12:29:58 +0200 |
| commit | 050dcbde0483c277cd632e05d8a0f73c87332785 (patch) | |
| tree | 4bd86b58929f564201f96bbdf73263bfb559648f /pipelines/stable_diffusion | |
| parent | Fix (diff) | |
| download | textual-inversion-diff-050dcbde0483c277cd632e05d8a0f73c87332785.tar.gz textual-inversion-diff-050dcbde0483c277cd632e05d8a0f73c87332785.tar.bz2 textual-inversion-diff-050dcbde0483c277cd632e05d8a0f73c87332785.zip | |
Update
Diffstat (limited to 'pipelines/stable_diffusion')
| -rw-r--r-- | pipelines/stable_diffusion/vlpn_stable_diffusion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pipelines/stable_diffusion/vlpn_stable_diffusion.py b/pipelines/stable_diffusion/vlpn_stable_diffusion.py index a0dff54..aa3dbc6 100644 --- a/pipelines/stable_diffusion/vlpn_stable_diffusion.py +++ b/pipelines/stable_diffusion/vlpn_stable_diffusion.py | |||
| @@ -301,7 +301,7 @@ class VlpnStableDiffusion(DiffusionPipeline): | |||
| 301 | init_timestep = min(int(num_inference_steps * strength), num_inference_steps) | 301 | init_timestep = min(int(num_inference_steps * strength), num_inference_steps) |
| 302 | 302 | ||
| 303 | t_start = max(num_inference_steps - init_timestep, 0) | 303 | t_start = max(num_inference_steps - init_timestep, 0) |
| 304 | timesteps = self.scheduler.timesteps[t_start:] | 304 | timesteps = self.scheduler.timesteps[t_start * self.scheduler.order:] |
| 305 | 305 | ||
| 306 | timesteps = timesteps.to(device) | 306 | timesteps = timesteps.to(device) |
| 307 | 307 | ||