fix bugs(admas timedeltas)

2025-05-20 12:25:50 +08:00
parent 45d5e8db5b
commit 4d4b3262c3
2 changed files with 2 additions and 2 deletions
--- a/src/diffusion/stateful_flow_matching/adam_sampling.py
+++ b/src/diffusion/stateful_flow_matching/adam_sampling.py
@@ -61,7 +61,7 @@ class AdamLMSampler(BaseSampler):
        timesteps = torch.linspace(0.0, 1 - self.last_step, self.num_steps)
        timesteps = torch.cat([timesteps, torch.tensor([1.0])], dim=0)
        self.timesteps = shift_respace_fn(timesteps, timeshift)
-        self.timedeltas = timesteps[1:] - self.timesteps[:-1]
+        self.timedeltas = self.timesteps[1:] - self.timesteps[:-1]
        self._reparameterize_coeffs()

    def _reparameterize_coeffs(self):