From 050dcbde0483c277cd632e05d8a0f73c87332785 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Thu, 20 Apr 2023 12:29:58 +0200 Subject: Update --- pipelines/stable_diffusion/vlpn_stable_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'pipelines') 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): init_timestep = min(int(num_inference_steps * strength), num_inference_steps) t_start = max(num_inference_steps - init_timestep, 0) - timesteps = self.scheduler.timesteps[t_start:] + timesteps = self.scheduler.timesteps[t_start * self.scheduler.order:] timesteps = timesteps.to(device) -- cgit v1.2.3-70-g09d2