Method
time_shift
(self, mu: float, sigma: float, t: torch.Tensor)
Source from the content-addressed store, hash-verified
| 337 | |
| 338 | # Copied from diffusers.schedulers.scheduling_flow_match_euler_discrete.set_timesteps |
| 339 | def time_shift(self, mu: float, sigma: float, t: torch.Tensor): |
| 340 | return math.exp(mu) / (math.exp(mu) + (1 / t - 1)**sigma) |
| 341 | |
| 342 | # Copied from diffusers.schedulers.scheduling_dpmsolver_multistep.DPMSolverMultistepScheduler.convert_model_output |
| 343 | def convert_model_output( |
Tested by
no test coverage detected