(
t,
shift=5.0,
num_timesteps=1000,
)
| 90 | |
| 91 | |
| 92 | def timestep_transform( |
| 93 | t, |
| 94 | shift=5.0, |
| 95 | num_timesteps=1000, |
| 96 | ): |
| 97 | t = t / num_timesteps |
| 98 | # shift the timestep based on ratio |
| 99 | new_t = shift * t / (1 + (shift - 1) * t) |
| 100 | new_t = new_t * num_timesteps |
| 101 | return new_t |
| 102 | |
| 103 | |
| 104 |