summaryrefslogtreecommitdiffstats
path: root/pipelines/stable_diffusion
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-04-20 12:29:58 +0200
committerVolpeon <git@volpeon.ink>2023-04-20 12:29:58 +0200
commit050dcbde0483c277cd632e05d8a0f73c87332785 (patch)
tree4bd86b58929f564201f96bbdf73263bfb559648f /pipelines/stable_diffusion
parentFix (diff)
downloadtextual-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.py2
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