From 96e7a92a7ed3aafd67d98e0c47ca94f3e43d3cd6 Mon Sep 17 00:00:00 2001 From: Cursor Agent Date: Sun, 20 Sep 2026 11:11:23 +0000 Subject: [PATCH] =?UTF-8?q?[fork=20demo=20=E2=80=94=20not=20for=20upstream?= =?UTF-8?q?]=20Fix=20EulerLite=20set=5Ftimesteps=20torch.flip?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Replace tensor [::-1] (unsupported) with torch.flip. Leave step() as TODO. Co-authored-by: ale93.moro --- src/diffusers/schedulers/scheduling_euler_lite.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/diffusers/schedulers/scheduling_euler_lite.py b/src/diffusers/schedulers/scheduling_euler_lite.py index 6f497f343..3e73022b8 100644 --- a/src/diffusers/schedulers/scheduling_euler_lite.py +++ b/src/diffusers/schedulers/scheduling_euler_lite.py @@ -36,7 +36,8 @@ def set_timesteps(self, num_inference_steps: int, device: Union[str, torch.devic """ self.num_inference_steps = num_inference_steps step = self.config.num_train_timesteps // num_inference_steps - timesteps = (torch.arange(0, num_inference_steps) * step).round()[::-1].clone() + timesteps = (torch.arange(0, num_inference_steps) * step).round().long() + timesteps = torch.flip(timesteps, dims=[0]) # torch has no [::-1] self.timesteps = timesteps.to(device) if device is not None else timesteps def step(