AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Convert sigmas to np.array in FlowMatch set_timesteps (#10088)

Browse Source
This commit is contained in:
hlky
2024-12-03 00:18:40 +00:00
committed by GitHub
parent 2312b27f79
commit 30f2e9bd20
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
View File

@@ -207,6 +207,7 @@ class FlowMatchEulerDiscreteScheduler(SchedulerMixin, ConfigMixin):
sigmas = timesteps / self.config.num_train_timesteps
else:
sigmas = np.array(sigmas).astype(np.float32)
num_inference_steps = len(sigmas)
self.num_inference_steps = num_inference_steps