Update src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py

a-r-r-o-w · yiyixuxu · web-flow · commit 2f12b7a196d5 · 2025-02-15T03:12:43.000+05:30
Co-authored-by: YiYi Xu &lt;yixu310@gmail.com&gt;
diff --git a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
@@ -300,7 +300,6 @@ def set_timesteps(
                     self._sigma_to_t(self.sigma_max), self._sigma_to_t(self.sigma_min), num_inference_steps
                 )
             sigmas = timesteps / self.config.num_train_timesteps
-            num_inference_steps = len(sigmas)
         else:
             sigmas = np.array(sigmas).astype(np.float32)
             num_inference_steps = len(sigmas)

Original file line number	Diff line number	Diff line change
`@@ -300,7 +300,6 @@ def set_timesteps(`
`300`	`300`	`self._sigma_to_t(self.sigma_max), self._sigma_to_t(self.sigma_min), num_inference_steps`
`301`	`301`	`)`
`302`	`302`	`sigmas = timesteps / self.config.num_train_timesteps`
`303`		`- num_inference_steps = len(sigmas)`
`304`	`303`	`else:`
`305`	`304`	`sigmas = np.array(sigmas).astype(np.float32)`
`306`	`305`	`num_inference_steps = len(sigmas)`